mridc.collections.reconstruction.models.varnet package

Submodules

mridc.collections.reconstruction.models.varnet.vn_block module

class mridc.collections.reconstruction.models.varnet.vn_block.VarNetBlock(model: Module, fft_centered: bool = True, fft_normalization: str = 'ortho', spatial_dims: Optional[Tuple[int, int]] = None, coil_dim: int = 1, no_dc: bool = False)[source]

Bases: Module

Model block for end-to-end variational network.

This model applies a combination of soft data consistency with the input model as a regularizer. A series of these blocks can be stacked to form the full variational network.

forward(pred: Tensor, ref_kspace: Tensor, sens_maps: Tensor, mask: Tensor) → Tensor[source]
Parameters

  • pred (Input data.) –

  • ref_kspace (Reference k-space data.) –

  • sens_maps (Coil sensitivity maps.) –

  • mask (Mask to apply to the data.) –

Return type

Reconstructed image.

sens_expand(x: Tensor, sens_maps: Tensor) → Tensor[source]

Expand the sensitivity maps to the same size as the input.

Parameters

  • x (Input data.) –

  • sens_maps (Coil Sensitivity maps.) –

Return type

SENSE reconstruction expanded to the same size as the input sens_maps.

sens_reduce(x: Tensor, sens_maps: Tensor) → Tensor[source]

Reduce the sensitivity maps.

Parameters

  • x (Input data.) –

  • sens_maps (Coil Sensitivity maps.) –

Return type

SENSE coil-combined reconstruction.

training: bool

Module contents