mridc.collections.reconstruction package

Subpackages

• mridc.collections.reconstruction.data package
  • Submodules
  • mridc.collections.reconstruction.data.mri_data module
    • FastMRICombinedSliceDataset
    • FastMRISliceDataset
      • FastMRISliceDataset.get_consecutive_slices()
    • et_query()
  • mridc.collections.reconstruction.data.subsample module
    • Equispaced1DMaskFunc
      • Equispaced1DMaskFunc.__call__()
    • Equispaced2DMaskFunc
      • Equispaced2DMaskFunc.__call__()
    • Gaussian1DMaskFunc
      • Gaussian1DMaskFunc.__call__()
      • Gaussian1DMaskFunc.gaussian_coordinates()
      • Gaussian1DMaskFunc.gaussian_kernel()
      • Gaussian1DMaskFunc.gaussian_kspace()
    • Gaussian2DMaskFunc
      • Gaussian2DMaskFunc.__call__()
      • Gaussian2DMaskFunc.gaussian_coordinates()
      • Gaussian2DMaskFunc.gaussian_kernel()
      • Gaussian2DMaskFunc.gaussian_kspace()
    • MaskFunc
      • MaskFunc.__call__()
      • MaskFunc.choose_acceleration()
    • Poisson2DMaskFunc
      • Poisson2DMaskFunc.__call__()
      • Poisson2DMaskFunc.centered_circle()
      • Poisson2DMaskFunc.poisson_disc2d()
    • RandomMaskFunc
      • RandomMaskFunc.__call__()
    • create_mask_for_mask_type()
    • temp_seed()
  • Module contents
• mridc.collections.reconstruction.metrics package
  • Submodules
  • mridc.collections.reconstruction.metrics.evaluate module
    • Metrics
      • Metrics.__repr__()
      • Metrics.means()
      • Metrics.push()
      • Metrics.stddevs()
    • evaluate()
    • mse()
    • nmse()
    • psnr()
    • ssim()
  • Module contents
• mridc.collections.reconstruction.models package
  • Subpackages
    • mridc.collections.reconstruction.models.cascadenet package
      • Submodules
      • mridc.collections.reconstruction.models.cascadenet.ccnn_block module
      • Module contents
    • mridc.collections.reconstruction.models.conv package
      • Submodules
      • mridc.collections.reconstruction.models.conv.conv2d module
      • mridc.collections.reconstruction.models.conv.gruconv2d module
      • Module contents
    • mridc.collections.reconstruction.models.convrecnet package
      • Submodules
      • mridc.collections.reconstruction.models.convrecnet.crnn_block module
      • Module contents
    • mridc.collections.reconstruction.models.crossdomain package
      • Submodules
      • mridc.collections.reconstruction.models.crossdomain.crossdomain module
      • mridc.collections.reconstruction.models.crossdomain.multicoil module
      • Module contents
    • mridc.collections.reconstruction.models.didn package
      • Submodules
      • mridc.collections.reconstruction.models.didn.didn module
      • Module contents
    • mridc.collections.reconstruction.models.multidomain package
      • Submodules
      • mridc.collections.reconstruction.models.multidomain.multidomain module
      • Module contents
    • mridc.collections.reconstruction.models.mwcnn package
      • Submodules
      • mridc.collections.reconstruction.models.mwcnn.mwcnn module
      • Module contents
    • mridc.collections.reconstruction.models.primaldual package
      • Submodules
      • mridc.collections.reconstruction.models.primaldual.pd module
      • Module contents
    • mridc.collections.reconstruction.models.recurrentvarnet package
      • Submodules
      • mridc.collections.reconstruction.models.recurrentvarnet.conv2gru module
      • mridc.collections.reconstruction.models.recurrentvarnet.recurentvarnet module
      • Module contents
    • mridc.collections.reconstruction.models.rim package
      • Submodules
      • mridc.collections.reconstruction.models.rim.conv_layers module
      • mridc.collections.reconstruction.models.rim.rim_block module
      • mridc.collections.reconstruction.models.rim.rnn_cells module
      • mridc.collections.reconstruction.models.rim.utils module
      • Module contents
    • mridc.collections.reconstruction.models.sigmanet package
      • Submodules
      • mridc.collections.reconstruction.models.sigmanet.dc_layers module
      • mridc.collections.reconstruction.models.sigmanet.sensitivity_net module
      • Module contents
    • mridc.collections.reconstruction.models.unet_base package
      • Submodules
      • mridc.collections.reconstruction.models.unet_base.unet_block module
      • Module contents
    • mridc.collections.reconstruction.models.variablesplittingnet package
      • Submodules
      • mridc.collections.reconstruction.models.variablesplittingnet.vsnet_block module
      • Module contents
    • mridc.collections.reconstruction.models.varnet package
      • Submodules
      • mridc.collections.reconstruction.models.varnet.vn_block module
      • Module contents
  • Submodules
  • mridc.collections.reconstruction.models.base module
    • BaseMRIReconstructionModel
      • BaseMRIReconstructionModel.log_image()
      • BaseMRIReconstructionModel.process_inputs()
      • BaseMRIReconstructionModel.process_loss()
      • BaseMRIReconstructionModel.setup_test_data()
      • BaseMRIReconstructionModel.setup_training_data()
      • BaseMRIReconstructionModel.setup_validation_data()
      • BaseMRIReconstructionModel.test_epoch_end()
      • BaseMRIReconstructionModel.test_step()
      • BaseMRIReconstructionModel.training
      • BaseMRIReconstructionModel.training_step()
      • BaseMRIReconstructionModel.validation_epoch_end()
      • BaseMRIReconstructionModel.validation_step()
    • BaseSensitivityModel
      • BaseSensitivityModel.batch_chans_to_chan_dim()
      • BaseSensitivityModel.chans_to_batch_dim()
      • BaseSensitivityModel.divide_root_sum_of_squares()
      • BaseSensitivityModel.forward()
      • BaseSensitivityModel.get_pad_and_num_low_freqs()
      • BaseSensitivityModel.training
    • DistributedMetricSum
      • DistributedMetricSum.compute()
      • DistributedMetricSum.update()
  • mridc.collections.reconstruction.models.ccnn module
    • CascadeNet
      • CascadeNet.allow_zero_length_dataloader_with_multiple_devices
      • CascadeNet.forward()
      • CascadeNet.mse_vals
      • CascadeNet.nmse_vals
      • CascadeNet.precision
      • CascadeNet.prepare_data_per_node
      • CascadeNet.psnr_vals
      • CascadeNet.ssim_vals
      • CascadeNet.training
  • mridc.collections.reconstruction.models.cirim module
    • CIRIM
      • CIRIM.allow_zero_length_dataloader_with_multiple_devices
      • CIRIM.forward()
      • CIRIM.mse_vals
      • CIRIM.nmse_vals
      • CIRIM.precision
      • CIRIM.prepare_data_per_node
      • CIRIM.process_intermediate_pred()
      • CIRIM.process_loss()
      • CIRIM.psnr_vals
      • CIRIM.ssim_vals
      • CIRIM.training
  • mridc.collections.reconstruction.models.crnn module
    • CRNNet
      • CRNNet.allow_zero_length_dataloader_with_multiple_devices
      • CRNNet.forward()
      • CRNNet.mse_vals
      • CRNNet.nmse_vals
      • CRNNet.precision
      • CRNNet.prepare_data_per_node
      • CRNNet.process_intermediate_pred()
      • CRNNet.process_loss()
      • CRNNet.psnr_vals
      • CRNNet.ssim_vals
      • CRNNet.training
  • mridc.collections.reconstruction.models.dunet module
    • DUNet
      • DUNet.allow_zero_length_dataloader_with_multiple_devices
      • DUNet.forward()
      • DUNet.mse_vals
      • DUNet.nmse_vals
      • DUNet.precision
      • DUNet.prepare_data_per_node
      • DUNet.psnr_vals
      • DUNet.ssim_vals
      • DUNet.training
  • mridc.collections.reconstruction.models.jointicnet module
    • JointICNet
      • JointICNet.allow_zero_length_dataloader_with_multiple_devices
      • JointICNet.forward()
      • JointICNet.mse_vals
      • JointICNet.nmse_vals
      • JointICNet.precision
      • JointICNet.prepare_data_per_node
      • JointICNet.psnr_vals
      • JointICNet.ssim_vals
      • JointICNet.training
      • JointICNet.update_C()
      • JointICNet.update_X()
  • mridc.collections.reconstruction.models.kikinet module
    • KIKINet
      • KIKINet.allow_zero_length_dataloader_with_multiple_devices
      • KIKINet.forward()
      • KIKINet.mse_vals
      • KIKINet.nmse_vals
      • KIKINet.precision
      • KIKINet.prepare_data_per_node
      • KIKINet.psnr_vals
      • KIKINet.ssim_vals
      • KIKINet.training
  • mridc.collections.reconstruction.models.lpd module
    • LPDNet
      • LPDNet.allow_zero_length_dataloader_with_multiple_devices
      • LPDNet.forward()
      • LPDNet.mse_vals
      • LPDNet.nmse_vals
      • LPDNet.precision
      • LPDNet.prepare_data_per_node
      • LPDNet.psnr_vals
      • LPDNet.ssim_vals
      • LPDNet.training
  • mridc.collections.reconstruction.models.multidomainnet module
    • MultiDomainNet
      • MultiDomainNet.allow_zero_length_dataloader_with_multiple_devices
      • MultiDomainNet.forward()
      • MultiDomainNet.mse_vals
      • MultiDomainNet.nmse_vals
      • MultiDomainNet.precision
      • MultiDomainNet.prepare_data_per_node
      • MultiDomainNet.psnr_vals
      • MultiDomainNet.ssim_vals
      • MultiDomainNet.training
  • mridc.collections.reconstruction.models.pics module
    • PICS
      • PICS.allow_zero_length_dataloader_with_multiple_devices
      • PICS.forward()
      • PICS.mse_vals
      • PICS.nmse_vals
      • PICS.precision
      • PICS.prepare_data_per_node
      • PICS.psnr_vals
      • PICS.ssim_vals
      • PICS.test_step()
      • PICS.training
  • mridc.collections.reconstruction.models.rvn module
    • RecurrentVarNet
      • RecurrentVarNet.allow_zero_length_dataloader_with_multiple_devices
      • RecurrentVarNet.forward()
      • RecurrentVarNet.mse_vals
      • RecurrentVarNet.nmse_vals
      • RecurrentVarNet.precision
      • RecurrentVarNet.prepare_data_per_node
      • RecurrentVarNet.psnr_vals
      • RecurrentVarNet.ssim_vals
      • RecurrentVarNet.training
  • mridc.collections.reconstruction.models.unet module
    • UNet
      • UNet.allow_zero_length_dataloader_with_multiple_devices
      • UNet.forward()
      • UNet.mse_vals
      • UNet.nmse_vals
      • UNet.precision
      • UNet.prepare_data_per_node
      • UNet.psnr_vals
      • UNet.ssim_vals
      • UNet.training
  • mridc.collections.reconstruction.models.vn module
    • VarNet
      • VarNet.allow_zero_length_dataloader_with_multiple_devices
      • VarNet.forward()
      • VarNet.mse_vals
      • VarNet.nmse_vals
      • VarNet.precision
      • VarNet.prepare_data_per_node
      • VarNet.psnr_vals
      • VarNet.ssim_vals
      • VarNet.training
  • mridc.collections.reconstruction.models.vsnet module
    • VSNet
      • VSNet.allow_zero_length_dataloader_with_multiple_devices
      • VSNet.forward()
      • VSNet.mse_vals
      • VSNet.nmse_vals
      • VSNet.precision
      • VSNet.prepare_data_per_node
      • VSNet.psnr_vals
      • VSNet.ssim_vals
      • VSNet.training
  • mridc.collections.reconstruction.models.xpdnet module
    • XPDNet
      • XPDNet.allow_zero_length_dataloader_with_multiple_devices
      • XPDNet.forward()
      • XPDNet.mse_vals
      • XPDNet.nmse_vals
      • XPDNet.precision
      • XPDNet.prepare_data_per_node
      • XPDNet.psnr_vals
      • XPDNet.ssim_vals
      • XPDNet.training
  • mridc.collections.reconstruction.models.zf module
    • ZF
      • ZF.allow_zero_length_dataloader_with_multiple_devices
      • ZF.forward()
      • ZF.mse_vals
      • ZF.nmse_vals
      • ZF.precision
      • ZF.prepare_data_per_node
      • ZF.psnr_vals
      • ZF.ssim_vals
      • ZF.test_step()
      • ZF.training
  • Module contents
• mridc.collections.reconstruction.parts package
  • Submodules
  • mridc.collections.reconstruction.parts.transforms module
    • MRIDataTransforms
      • MRIDataTransforms.__call__()
  • mridc.collections.reconstruction.parts.utils module
    • apply_mask()
    • batched_mask_center()
    • center_crop()
    • center_crop_to_smallest()
    • complex_center_crop()
    • mask_center()
  • Module contents

Module contents