N2V Module

Instantiate N2VModule.

Parameters:

• algorithm_config (N2VAlgorithm or dict) –

  Configuration for the N2V algorithm, either as an N2VAlgorithm instance or a dictionary.

Configure optimizer and learning rate scheduler.

Returns:

• dict[str, Any] –

  A dictionary containing the optimizer and learning rate scheduler.

Forward pass.

Parameters:

• x (Tensor) –

  Input tensor.

Returns:

• Tensor –

  Model output tensor.

On fit start hook for N2V module.

Prediction step for N2V model.

Parameters:

• batch (ImageRegionData or (ImageRegionData, ImageRegionData)) –

  A tuple containing the input data and optionally the target data.

• batch_idx (int) –

  The index of the current batch in the prediction loop.

Returns:

• ImageRegionData –

  The output batch containing the predictions.

Training step for N2V model.

Parameters:

• batch (ImageRegionData or (ImageRegionData, ImageRegionData)) –

  A tuple containing the input data and the target data.

• batch_idx (int) –

  The index of the current batch in the training loop.

Returns:

• Tensor –

  The loss value for the current training step.

Validation step for N2V model.

Parameters:

• batch (ImageRegionData or (ImageRegionData, ImageRegionData)) –

  A tuple containing the input data and the target data.

• batch_idx (int) –

  The index of the current batch in the validation loop.