Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | R | S | T | U | V | W | X | Z A accelerations (direct.common.subsample_config.MaskingConfig attribute) ACS_MASK (direct.types.TransformKey attribute) ActivationType (class in direct.nn.types) add_graph() (direct.utils.events.EventStorage method) add_image() (direct.utils.events.EventStorage method) add_scalar() (direct.utils.events.EventStorage method) add_scalars() (direct.utils.events.EventStorage method) AddBooleanKeysModule (class in direct.data.mri_transforms) additional_models (direct.config.defaults.DefaultConfig attribute) Algorithm (class in direct.algorithms.optimization) all_gather() (in module direct.utils.communication) apply_mask() (in module direct.data.transforms) apply_padding() (in module direct.data.transforms) ApplyMaskModule (class in direct.data.mri_transforms) ApplyZeroPadding (class in direct.data.mri_transforms) AREA (direct.data.mri_transforms.RescaleMode attribute) Args (class in direct.environment) ArrayEncoder (class in direct.utils.io) assert_complex() (in module direct.utils.asserts) assert_positive_integer() (in module direct.utils.asserts) assert_same_shape() (in module direct.utils.asserts) augmentation (direct.data.datasets_config.TransformsConfig attribute) AugmentationTransformConfig (class in direct.data.datasets_config) auxiliary_steps (direct.nn.vsharp.config.VSharpNet3DConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) avg() (direct.utils.events.HistoryBuffer method) B B0 (direct.data.datasets_config.SheppLoganDatasetConfig attribute) B_op() (direct.nn.conjgradnet.conjgrad.ConjGrad method) backward_operator (direct.config.defaults.PhysicsConfig attribute) BAN (direct.nn.conjgradnet.conjgrad.CGUpdateType attribute) BASE_URL (direct.common.subsample.CalgaryCampinasMaskFunc attribute) BaseArgs (class in direct.cli.upload) BaseConfig (class in direct.config) BaseMaskFunc (class in direct.common.subsample) batch_psnr() (in module direct.functionals) (in module direct.functionals.psnr) batch_sampler() (direct.data.samplers.ConcatDatasetBatchSampler method) batch_size (direct.config.defaults.InferenceConfig attribute) (direct.config.defaults.TrainingConfig attribute) (direct.config.defaults.ValidationConfig attribute) batchnorm (direct.nn.resnet.config.ResNetConfig attribute) BatchVolumeSampler (class in direct.data.samplers) BICUBIC (direct.data.mri_transforms.RescaleMode attribute) BILINEAR (direct.data.mri_transforms.RescaleMode attribute) BOTH (direct.data.mri_transforms.RandomFlipType attribute) build_batch_sampler() (direct.engine.Engine static method) build_dataset_from_input() (in module direct.data.datasets) build_inference_transforms() (in module direct.inference) build_loader() (direct.engine.Engine static method) build_loss() (direct.engine.Engine method) (direct.nn.mri_models.MRIModelEngine method) build_masking_function() (in module direct.common.subsample) build_metrics() (direct.engine.Engine method) build_mri_transforms() (in module direct.data.mri_transforms) build_operators() (in module direct.environment) build_post_mri_transforms() (in module direct.data.mri_transforms) build_pre_mri_transforms() (in module direct.data.mri_transforms) build_regularizers() (direct.engine.Engine method) build_supervised_mri_transforms() (in module direct.data.mri_transforms) build_training_datasets_from_environment() (in module direct.train) build_transforms_from_environment() (in module direct.train) C calculate_md5() (in module direct.utils.io) calculate_sensitivity_map() (direct.algorithms.mri_algorithms.EspiritCalibration method) calgary_campinas_psnr() (in module direct.functionals) (in module direct.functionals.challenges) calgary_campinas_ssim() (in module direct.functionals) (in module direct.functionals.challenges) calgary_campinas_vif() (in module direct.functionals) (in module direct.functionals.challenges) CalgaryCampinasConfig (class in direct.data.datasets_config) CalgaryCampinasDataset (class in direct.data.datasets) CalgaryCampinasMaskFunc (class in direct.common.subsample) CartesianEquispacedMaskFunc (class in direct.common.subsample) CartesianMagicMaskFunc (class in direct.common.subsample) CartesianRandomMaskFunc (class in direct.common.subsample) CartesianVerticalMaskFunc (class in direct.common.subsample) cast_as_path() (in module direct.utils) center_crop() (in module direct.data.transforms) center_fractions (direct.common.subsample_config.MaskingConfig attribute) center_mask_func() (direct.common.subsample.CartesianVerticalMaskFunc static method) centered_disk_mask() (in module direct.common.subsample) cg() (direct.nn.conjgradnet.conjgrad.ConjGrad method) cg_iters (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) cg_param_update_type (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) cg_tol (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) CGUpdateType (class in direct.nn.conjgradnet.conjgrad) check_integrity() (in module direct.utils.io) check_is_valid_url() (in module direct.utils.io) check_md5() (in module direct.utils.io) check_train_val() (in module direct.cli.utils) checkpoint_and_write_to_logs() (direct.engine.Engine method) checkpoint_model_at_interval() (direct.engine.Engine method) checkpoint_steps (direct.config.defaults.CheckpointerConfig attribute) Checkpointer (class in direct.checkpointer) checkpointer (direct.config.defaults.TrainingConfig attribute) CheckpointerConfig (class in direct.config.defaults) choose_acceleration() (direct.common.subsample.BaseMaskFunc method) chunks() (in module direct.utils) circular_centered_mask() (direct.common.subsample.CalgaryCampinasMaskFunc static method) (direct.common.subsample.CIRCUSMaskFunc static method) CIRCUS_RADIAL (direct.common.subsample.CIRCUSSamplingMode attribute) circus_radial_mask() (direct.common.subsample.CIRCUSMaskFunc method) CIRCUS_SPIRAL (direct.common.subsample.CIRCUSSamplingMode attribute) circus_spiral_mask() (direct.common.subsample.CIRCUSMaskFunc method) CIRCUSMaskFunc (class in direct.common.subsample) CIRCUSSamplingMode (class in direct.common.subsample) CIRIM (class in direct.nn.cirim.cirim) CIRIMConfig (class in direct.nn.cirim.config) CIRIMEngine (class in direct.nn.cirim.cirim_engine) clear_images() (direct.utils.events.EventStorage method) close() (direct.utils.events.EventWriter method) (direct.utils.events.JSONWriter method) (direct.utils.events.TensorboardWriter method) CMRxReconConfig (class in direct.data.datasets_config) CMRxReconDataset (class in direct.data.datasets) CommonMetricPrinter (class in direct.utils.events) COMPLEX (direct.data.mri_transforms.ReconstructionType attribute) complex_bmm() (in module direct.data.transforms) complex_center_crop() (in module direct.data.transforms) complex_division() (in module direct.data.transforms) complex_dot_product() (in module direct.data.transforms) complex_image_resize() (in module direct.data.transforms) complex_mm() (in module direct.data.transforms) COMPLEX_MOD (direct.data.mri_transforms.ReconstructionType attribute) complex_multiplication() (in module direct.data.transforms) complex_random_crop() (in module direct.data.transforms) complex_whiten() (direct.data.mri_transforms.WhitenDataModule method) Compose (class in direct.data.mri_transforms) compress_coils (direct.data.datasets_config.TransformsConfig attribute) CompressCoilModule (class in direct.data.mri_transforms) compute_loss_on_data() (direct.nn.mri_models.MRIModelEngine method) compute_mask (direct.data.datasets_config.CMRxReconConfig attribute) compute_model_per_coil() (direct.nn.lpd.lpd.DualNet static method) (direct.nn.mri_models.MRIModelEngine method) (direct.nn.varsplitnet.varsplitnet.MRIVarSplitNet method) compute_per_coil (direct.nn.iterdualnet.config.IterDualNetConfig attribute) compute_sense_init() (direct.nn.recurrentvarnet.recurrentvarnet.RecurrentVarNet method) (direct.nn.rim.rim.RIM method) (direct.nn.unet.unet_2d.Unet2d method) compute_sensitivity_map() (direct.nn.mri_models.MRIModelEngine method) ComputeImageModule (class in direct.data.mri_transforms) ComputeScalingFactorModule (class in direct.data.mri_transforms) ComputeZeroPadding (class in direct.data.mri_transforms) ConcatDataset (class in direct.data.datasets) ConcatDatasetBatchSampler (class in direct.data.samplers) ConjGrad (class in direct.nn.conjgradnet.conjgrad) ConjGradNet (class in direct.nn.conjgradnet.conjgradnet) ConjGradNetConfig (class in direct.nn.conjgradnet.config) ConjGradNetEngine (class in direct.nn.conjgradnet.conjgradnet_engine) conjugate() (in module direct.data.transforms) CONV (direct.nn.types.ModelName attribute) Conv2d (class in direct.nn.conv.conv) Conv2dGRU (class in direct.nn.recurrent.recurrent) conv_activation (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) conv_batchnorm (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) conv_hidden_channels (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) conv_n_convs (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) ConvBlock (class in direct.nn.mwcnn.mwcnn) (class in direct.nn.unet.unet_2d) ConvBlock3D (class in direct.nn.unet.unet_3d) ConvNonlinear (class in direct.nn.cirim.cirim) ConvRNNStack (class in direct.nn.cirim.cirim) count_parameters() (in module direct.utils) CreateSamplingMask (class in direct.data.mri_transforms) crop (direct.config.defaults.InferenceConfig attribute) (direct.config.defaults.LossConfig attribute) (direct.config.defaults.ValidationConfig attribute) (direct.data.datasets_config.CropTransformConfig attribute) crop_center() (direct.common.subsample.KtBaseMaskFunc static method) crop_outer_slices (direct.data.datasets_config.CalgaryCampinasConfig attribute) crop_to_acs() (in module direct.data.transforms) crop_to_bbox() (in module direct.data.bbox) (in module direct.utils.bbox) crop_to_largest() (in module direct.data.bbox) (in module direct.utils.bbox) crop_to_shape() (direct.nn.didn.didn.DIDN static method) (direct.nn.didn.didn.DUB static method) (direct.nn.mwcnn.mwcnn.MWCNN static method) crop_type (direct.data.datasets_config.CropTransformConfig attribute) CropKspace (class in direct.data.mri_transforms) cropping (direct.data.datasets_config.TransformsConfig attribute) CropTransformConfig (class in direct.data.datasets_config) CrossDomainNetwork (class in direct.nn.crossdomain.crossdomain) cumsum() (direct.data.datasets.ConcatDataset static method) (direct.data.samplers.ConcatDatasetBatchSampler static method) D data_dict (direct.engine.DoIterationOutput attribute) data_root (direct.data.datasets_config.CMRxReconConfig attribute) (direct.data.datasets_config.H5SliceConfig attribute) DataDimensionality (class in direct.engine) dataset (direct.config.defaults.InferenceConfig attribute) DatasetConfig (class in direct.data.datasets_config) datasets (direct.config.defaults.TrainingConfig attribute) (direct.config.defaults.ValidationConfig attribute) default() (direct.utils.io.ArrayEncoder method) default_mr_ellipsoid_parameters() (direct.data.datasets.SheppLoganDataset static method) DEFAULT_NUM_ELLIPSOIDS (direct.data.datasets.SheppLoganDataset attribute) DefaultConfig (class in direct.config.defaults) delete_acs_mask (direct.data.datasets_config.TransformsConfig attribute) delete_kspace (direct.data.datasets_config.TransformsConfig attribute) DeleteKeysModule (class in direct.data.mri_transforms) denoiser_architecture (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) dense_connect (direct.nn.rim.config.RIMConfig attribute) depth (direct.nn.cirim.config.CIRIMConfig attribute) (direct.nn.rim.config.RIMConfig attribute) detach_dict() (in module direct.utils) DIAGONAL_LEFT (direct.ssl.ssl.HalfSplitType attribute) DIAGONAL_RIGHT (direct.ssl.ssl.HalfSplitType attribute) dict_flatten() (in module direct.utils) dict_to_device() (in module direct.utils) DIDN (class in direct.nn.didn.didn) (direct.nn.types.ModelName attribute) didn_hidden_channels (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) didn_num_convs_recon (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) didn_num_dubs (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) DilatedConvBlock (class in direct.nn.mwcnn.mwcnn) direct module direct.algorithms module direct.algorithms.mri_algorithms module direct.algorithms.optimization module direct.checkpointer module direct.cli module direct.cli.predict module direct.cli.train module direct.cli.upload module direct.cli.utils module direct.common module direct.common.subsample module direct.common.subsample_config module direct.config module direct.config.defaults module direct.constants module direct.data module direct.data.bbox module direct.data.datasets module direct.data.datasets_config module direct.data.fake module direct.data.h5_data module direct.data.lr_scheduler module direct.data.mri_transforms module direct.data.samplers module direct.data.sens module direct.data.transforms module direct.engine module direct.environment module direct.exceptions module direct.functionals module direct.functionals.challenges module direct.functionals.grad module direct.functionals.hfen module direct.functionals.nmae module direct.functionals.nmse module direct.functionals.psnr module direct.functionals.snr module direct.functionals.ssim module direct.inference module direct.launch module direct.nn module direct.nn.cirim module direct.nn.cirim.cirim module direct.nn.cirim.cirim_engine module direct.nn.cirim.config module direct.nn.conjgradnet module direct.nn.conjgradnet.config module direct.nn.conjgradnet.conjgrad module direct.nn.conjgradnet.conjgradnet module direct.nn.conjgradnet.conjgradnet_engine module direct.nn.conv module direct.nn.conv.conv module direct.nn.crossdomain module direct.nn.crossdomain.crossdomain module direct.nn.crossdomain.multicoil module direct.nn.didn module direct.nn.didn.didn module direct.nn.get_nn_model_config module direct.nn.iterdualnet module direct.nn.iterdualnet.config module direct.nn.iterdualnet.iterdualnet module direct.nn.iterdualnet.iterdualnet_engine module direct.nn.jointicnet module direct.nn.jointicnet.config module direct.nn.jointicnet.jointicnet module direct.nn.jointicnet.jointicnet_engine module direct.nn.kikinet module direct.nn.kikinet.config module direct.nn.kikinet.kikinet module direct.nn.kikinet.kikinet_engine module direct.nn.lpd module direct.nn.lpd.config module direct.nn.lpd.lpd module direct.nn.lpd.lpd_engine module direct.nn.mobilenet module direct.nn.mobilenet.config module direct.nn.mobilenet.mobilenet module direct.nn.mri_models module direct.nn.multidomainnet module direct.nn.multidomainnet.config module direct.nn.multidomainnet.multidomain module direct.nn.multidomainnet.multidomainnet module direct.nn.multidomainnet.multidomainnet_engine module direct.nn.mwcnn module direct.nn.mwcnn.mwcnn module direct.nn.recurrent module direct.nn.recurrent.recurrent module direct.nn.recurrentvarnet module direct.nn.recurrentvarnet.config module direct.nn.recurrentvarnet.recurrentvarnet module direct.nn.recurrentvarnet.recurrentvarnet_engine module direct.nn.resnet module direct.nn.resnet.config module direct.nn.resnet.resnet module direct.nn.rim module direct.nn.rim.config module direct.nn.rim.rim module direct.nn.rim.rim_engine module direct.nn.ssl module direct.nn.ssl.mri_models module direct.nn.types module direct.nn.unet module direct.nn.unet.config module direct.nn.unet.unet_2d module direct.nn.unet.unet_3d module direct.nn.unet.unet_engine module direct.nn.varnet module direct.nn.varnet.config module direct.nn.varnet.varnet module direct.nn.varnet.varnet_engine module direct.nn.varsplitnet module direct.nn.varsplitnet.config module direct.nn.varsplitnet.varsplitnet module direct.nn.varsplitnet.varsplitnet_engine module direct.nn.vsharp module direct.nn.vsharp.config module direct.nn.vsharp.vsharp module direct.nn.vsharp.vsharp_engine module direct.nn.xpdnet module direct.nn.xpdnet.config module direct.nn.xpdnet.xpdnet module direct.nn.xpdnet.xpdnet_engine module direct.predict module direct.ssl module direct.ssl.mask_fillers module direct.ssl.ssl module direct.train module direct.types module direct.utils module direct.utils.asserts module direct.utils.bbox module direct.utils.communication module direct.utils.dataset module direct.utils.events module direct.utils.imports module direct.utils.io module direct.utils.logging module direct.utils.models module direct.utils.writers module DirectEnum (class in direct.types) DirectException DirectModule (class in direct.utils) DirectTransform (class in direct.utils) DistributedSampler (class in direct.data.samplers) DistributedSequentialSampler (class in direct.data.samplers) DoIterationOutput (class in direct.engine) done() (direct.algorithms.optimization.Algorithm method) download_and_extract_archive() (in module direct.utils.io) download_url() (in module direct.utils.io) dropout_probability (direct.nn.multidomainnet.config.MultiDomainNetConfig attribute) (direct.nn.unet.config.NormUnetModel2dConfig attribute) (direct.nn.unet.config.Unet2dConfig attribute) (direct.nn.unet.config.UnetModel2dConfig attribute) (direct.nn.unet.config.UnetModel3dConfig attribute) dual_conv_batchnorm (direct.nn.lpd.config.LPDNetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) dual_conv_hidden_channels (direct.nn.lpd.config.LPDNetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) dual_conv_n_convs (direct.nn.lpd.config.LPDNetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) dual_didn_hidden_channels (direct.nn.lpd.config.LPDNetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) dual_didn_num_convs_recon (direct.nn.lpd.config.LPDNetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) dual_didn_num_dubs (direct.nn.lpd.config.LPDNetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) dual_model_architecture (direct.nn.lpd.config.LPDNetConfig attribute) dual_unet_dropout_probability (direct.nn.lpd.config.LPDNetConfig attribute) dual_unet_num_filters (direct.nn.lpd.config.LPDNetConfig attribute) dual_unet_num_pool_layers (direct.nn.lpd.config.LPDNetConfig attribute) DualNet (class in direct.nn.lpd.lpd) DUB (class in direct.nn.didn.didn) DWT (class in direct.nn.mwcnn.mwcnn) DY (direct.nn.conjgradnet.conjgrad.CGUpdateType attribute) DYNAMIC (direct.common.subsample.MaskFuncMode attribute) (direct.types.MaskFuncMode attribute) E ELLIPSOID_NUM_PARAMS (direct.data.datasets.SheppLoganDataset attribute) EndToEndVarNet (class in direct.nn.varnet.varnet) EndToEndVarNetBlock (class in direct.nn.varnet.varnet) EndToEndVarNetConfig (class in direct.nn.varnet.config) EndToEndVarNetEngine (class in direct.nn.varnet.varnet_engine) EndToEndVarNetJSSLEngine (class in direct.nn.varnet.varnet_engine) EndToEndVarNetSSLEngine (class in direct.nn.varnet.varnet_engine) Engine (class in direct.engine) engine_name (direct.config.defaults.ModelConfig attribute) ensure_list() (in module direct.utils) EquispacedMaskFunc (class in direct.common.subsample) ESPIRIT (direct.data.mri_transforms.SensitivityMapType attribute) EspiritCalibration (class in direct.algorithms.mri_algorithms) estimate_acs_image() (direct.data.mri_transforms.EstimateSensitivityMapModule method) estimate_body_coil_image (direct.data.datasets_config.TransformsConfig attribute) estimate_sensitivity_maps (direct.data.datasets_config.SensitivityMapEstimationTransformConfig attribute) EstimateBodyCoilImage (class in direct.data.mri_transforms) EstimateSensitivityMapModule (class in direct.data.mri_transforms) evaluate() (direct.engine.Engine method) (direct.nn.mri_models.MRIModelEngine method) evaluate_dict() (in module direct.utils) EventStorage (class in direct.utils.events) EventWriter (class in direct.utils.events) expand_operator() (in module direct.data.transforms) explicit_zero_padding() (direct.data.datasets.FastMRIDataset static method) extra_keys (direct.data.datasets_config.CMRxReconConfig attribute) extract_archive() (in module direct.utils.io) extract_names() (in module direct.environment) F FakeMRIBlobsConfig (class in direct.data.datasets_config) FakeMRIBlobsDataset (class in direct.data.datasets) FakeMRIData (class in direct.data.fake) fastmri_nmse() (in module direct.functionals) (in module direct.functionals.challenges) fastmri_psnr() (in module direct.functionals) (in module direct.functionals.challenges) fastmri_ssim() (in module direct.functionals) (in module direct.functionals.challenges) FastMRIConfig (class in direct.data.datasets_config) FastMRIDataset (class in direct.data.datasets) FastMRIEquispacedMaskFunc (class in direct.common.subsample) FastMRIMagicMaskFunc (class in direct.common.subsample) FastMRIRandomMaskFunc (class in direct.common.subsample) fft() (direct.data.datasets.SheppLoganDataset static method) (in module direct.data.fake) fft2() (in module direct.data.transforms) fftshift() (in module direct.data.transforms) file_or_url() (in module direct.cli.utils) filenames_filter (direct.data.datasets_config.CMRxReconConfig attribute) (direct.data.datasets_config.H5SliceConfig attribute) filenames_lists (direct.data.datasets_config.CMRxReconConfig attribute) (direct.data.datasets_config.H5SliceConfig attribute) filenames_lists_root (direct.data.datasets_config.CMRxReconConfig attribute) (direct.data.datasets_config.H5SliceConfig attribute) find_nearest_empty_location() (direct.common.subsample.KtBaseMaskFunc static method) fit() (direct.algorithms.optimization.Algorithm method) fix_state_dict_module_prefix() (in module direct.utils.models) forward() (direct.algorithms.mri_algorithms.EspiritCalibration method) (direct.data.mri_transforms.AddBooleanKeysModule method) (direct.data.mri_transforms.ApplyMaskModule method) (direct.data.mri_transforms.CompressCoilModule method) (direct.data.mri_transforms.ComputeImageModule method) (direct.data.mri_transforms.ComputeScalingFactorModule method) (direct.data.mri_transforms.DeleteKeysModule method) (direct.data.mri_transforms.EstimateSensitivityMapModule method) (direct.data.mri_transforms.NormalizeModule method) (direct.data.mri_transforms.PadCoilDimensionModule method) (direct.data.mri_transforms.RenameKeysModule method) (direct.data.mri_transforms.WhitenDataModule method) (direct.functionals.hfen.HFENLoss method) (direct.functionals.HFENLoss method) (direct.functionals.nmae.NMAELoss method) (direct.functionals.NMAELoss method) (direct.functionals.nmse.NMSELoss method) (direct.functionals.nmse.NRMSELoss method) (direct.functionals.NMSELoss method) (direct.functionals.NRMSELoss method) (direct.functionals.psnr.PSNRLoss method) (direct.functionals.PSNRLoss method) (direct.functionals.snr.SNRLoss method) (direct.functionals.SNRLoss method) (direct.functionals.ssim.SSIM3DLoss method) (direct.functionals.ssim.SSIMLoss method) (direct.functionals.SSIM3DLoss method) (direct.functionals.SSIMLoss method) (direct.nn.cirim.cirim.CIRIM method) (direct.nn.cirim.cirim.ConvNonlinear method) (direct.nn.cirim.cirim.ConvRNNStack method) (direct.nn.cirim.cirim.IndRNNCell method) (direct.nn.cirim.cirim.RIMBlock method) (direct.nn.conjgradnet.conjgrad.ConjGrad method) (direct.nn.conjgradnet.conjgradnet.ConjGradNet method) (direct.nn.conv.conv.Conv2d method) (direct.nn.crossdomain.crossdomain.CrossDomainNetwork method) (direct.nn.crossdomain.multicoil.MultiCoil method) (direct.nn.didn.didn.DIDN method) (direct.nn.didn.didn.DUB method) (direct.nn.didn.didn.ReconBlock method) (direct.nn.didn.didn.Subpixel method) (direct.nn.iterdualnet.iterdualnet.IterDualNet method) (direct.nn.jointicnet.jointicnet.JointICNet method) (direct.nn.kikinet.kikinet.KIKINet method) (direct.nn.lpd.lpd.DualNet method) (direct.nn.lpd.lpd.LPDNet method) (direct.nn.lpd.lpd.PrimalNet method) (direct.nn.mobilenet.mobilenet.MobileNetV2 method) (direct.nn.multidomainnet.multidomain.MultiDomainConv2d method) (direct.nn.multidomainnet.multidomain.MultiDomainConvBlock method) (direct.nn.multidomainnet.multidomain.MultiDomainConvTranspose2d method) (direct.nn.multidomainnet.multidomain.MultiDomainUnet2d method) (direct.nn.multidomainnet.multidomain.TransposeMultiDomainConvBlock method) (direct.nn.multidomainnet.multidomainnet.MultiDomainNet method) (direct.nn.multidomainnet.multidomainnet.StandardizationLayer method) (direct.nn.mwcnn.mwcnn.ConvBlock method) (direct.nn.mwcnn.mwcnn.DilatedConvBlock method) (direct.nn.mwcnn.mwcnn.DWT method) (direct.nn.mwcnn.mwcnn.IWT method) (direct.nn.mwcnn.mwcnn.MWCNN method) (direct.nn.recurrent.recurrent.Conv2dGRU method) (direct.nn.recurrent.recurrent.NormConv2dGRU method) (direct.nn.recurrentvarnet.recurrentvarnet.RecurrentInit method) (direct.nn.recurrentvarnet.recurrentvarnet.RecurrentVarNet method) (direct.nn.recurrentvarnet.recurrentvarnet.RecurrentVarNetBlock method) (direct.nn.resnet.resnet.ResNet method) (direct.nn.resnet.resnet.ResNetBlock method) (direct.nn.rim.rim.MRILogLikelihood method) (direct.nn.rim.rim.RIM method) (direct.nn.rim.rim.RIMInit method) (direct.nn.unet.unet_2d.ConvBlock method) (direct.nn.unet.unet_2d.NormUnetModel2d method) (direct.nn.unet.unet_2d.TransposeConvBlock method) (direct.nn.unet.unet_2d.Unet2d method) (direct.nn.unet.unet_2d.UnetModel2d method) (direct.nn.unet.unet_3d.ConvBlock3D method) (direct.nn.unet.unet_3d.NormUnetModel3d method) (direct.nn.unet.unet_3d.TransposeConvBlock3D method) (direct.nn.unet.unet_3d.UnetModel3d method) (direct.nn.varnet.varnet.EndToEndVarNet method) (direct.nn.varnet.varnet.EndToEndVarNetBlock method) (direct.nn.varsplitnet.varsplitnet.MRIVarSplitNet method) (direct.nn.vsharp.vsharp.LagrangeMultipliersInitializer method) (direct.nn.vsharp.vsharp.LagrangeMultipliersInitializer3D method) (direct.nn.vsharp.vsharp.VSharpNet method) (direct.nn.vsharp.vsharp.VSharpNet3D method) (direct.utils.DirectModule method) forward_function() (direct.nn.conjgradnet.conjgradnet_engine.ConjGradNetEngine method) (direct.nn.iterdualnet.iterdualnet_engine.IterDualNetEngine method) (direct.nn.jointicnet.jointicnet_engine.JointICNetEngine method) (direct.nn.kikinet.kikinet_engine.KIKINetEngine method) (direct.nn.lpd.lpd_engine.LPDNetEngine method) (direct.nn.mri_models.MRIModelEngine method) (direct.nn.multidomainnet.multidomainnet_engine.MultiDomainNetEngine method) (direct.nn.recurrentvarnet.recurrentvarnet_engine.RecurrentVarNetEngine method) (direct.nn.ssl.mri_models.SSLMRIModelEngine method) (direct.nn.unet.unet_engine.Unet2dEngine method) (direct.nn.unet.unet_engine.Unet2dJSSLEngine method) (direct.nn.unet.unet_engine.Unet2dSSLEngine method) (direct.nn.varnet.varnet_engine.EndToEndVarNetEngine method) (direct.nn.varnet.varnet_engine.EndToEndVarNetJSSLEngine method) (direct.nn.varnet.varnet_engine.EndToEndVarNetSSLEngine method) (direct.nn.varsplitnet.varsplitnet_engine.MRIVarSplitNetEngine method) (direct.nn.vsharp.vsharp_engine.VSharpNet3DEngine method) (direct.nn.vsharp.vsharp_engine.VSharpNetEngine method) (direct.nn.vsharp.vsharp_engine.VSharpNetJSSLEngine method) (direct.nn.vsharp.vsharp_engine.VSharpNetSSLEngine method) (direct.nn.xpdnet.xpdnet_engine.XPDNetEngine method) forward_operator (direct.config.defaults.PhysicsConfig attribute) FR (direct.nn.conjgradnet.conjgrad.CGUpdateType attribute) from_str() (direct.types.DirectEnum class method) function (direct.config.defaults.FunctionConfig attribute) FunctionConfig (class in direct.config.defaults) G gather() (in module direct.utils.communication) GAUSSIAN (direct.ssl.ssl.MaskSplitterType attribute) Gaussian1DMaskFunc (class in direct.common.subsample) Gaussian2DMaskFunc (class in direct.common.subsample) gaussian_fill() (in module direct.ssl.mask_fillers) GaussianMaskSplitterModule (class in direct.ssl.ssl) gen_bar_updater() (in module direct.utils.io) get_event_storage() (in module direct.utils.events) get_filenames_for_datasets() (in module direct.utils.dataset) get_filenames_for_datasets_from_config() (in module direct.utils.dataset) get_kspace() (direct.data.fake.FakeMRIData method) get_local_rank() (in module direct.utils.communication) get_local_size() (in module direct.utils.communication) get_lr() (direct.data.lr_scheduler.WarmupCosineLR method) (direct.data.lr_scheduler.WarmupMultiStepLR method) get_num_slices() (direct.data.h5_data.H5SliceData method) get_rank() (in module direct.utils.communication) get_root_of_file() (in module direct.train) get_slice_data() (direct.data.datasets.CMRxReconDataset method) (direct.data.h5_data.H5SliceData method) get_square_ordered_idxs() (direct.common.subsample.CIRCUSMaskFunc static method) get_world_size() (in module direct.utils.communication) git_hash() (in module direct.utils) global_avg() (direct.utils.events.HistoryBuffer method) GRAD_L1_LOSS (direct.nn.types.LossFunType attribute) GRAD_L2_LOSS (direct.nn.types.LossFunType attribute) gradient_clipping (direct.config.defaults.TrainingConfig attribute) gradient_debug (direct.config.defaults.TrainingConfig attribute) gradient_steps (direct.config.defaults.TrainingConfig attribute) GYROMAGNETIC_RATIO (direct.data.datasets.SheppLoganDataset attribute) H H5SliceConfig (class in direct.data.datasets_config) H5SliceData (class in direct.data.h5_data) HALF (direct.ssl.ssl.MaskSplitterType attribute) HalfMaskSplitterModule (class in direct.ssl.ssl) HalfSplitType (class in direct.ssl.ssl) hfen_l1() (in module direct.functionals) (in module direct.functionals.hfen) HFEN_L1_LOSS (direct.nn.types.LossFunType attribute) HFEN_L1_NORM_LOSS (direct.nn.types.LossFunType attribute) hfen_l2() (in module direct.functionals) (in module direct.functionals.hfen) HFEN_L2_LOSS (direct.nn.types.LossFunType attribute) HFEN_L2_NORM_LOSS (direct.nn.types.LossFunType attribute) HFENL1Loss (class in direct.functionals) (class in direct.functionals.hfen) HFENL2Loss (class in direct.functionals) (class in direct.functionals.hfen) HFENLoss (class in direct.functionals) (class in direct.functionals.hfen) hidden_channels (direct.nn.resnet.config.ResNetConfig attribute) (direct.nn.rim.config.RIMConfig attribute) histories() (direct.utils.events.EventStorage method) history() (direct.utils.events.EventStorage method) HistoryBuffer (class in direct.utils.events) HORIZONTAL (direct.data.mri_transforms.RandomFlipType attribute) (direct.ssl.ssl.HalfSplitType attribute) I IFFT (direct.data.mri_transforms.ReconstructionType attribute) ifft() (in module direct.data.fake) ifft2() (in module direct.data.transforms) ifftshift() (in module direct.data.transforms) image_center_crop (direct.data.datasets_config.CropTransformConfig attribute) image_conv_activation (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_conv_batchnorm (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_conv_hidden_channels (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_conv_n_convs (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_correction() (direct.nn.crossdomain.crossdomain.CrossDomainNetwork method) image_didn_hidden_channels (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_didn_num_convs_recon (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_didn_num_dubs (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_init (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) (direct.nn.resnet.config.ResNetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNet3DConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_initialization (direct.nn.rim.config.RIMConfig attribute) (direct.nn.unet.config.Unet2dConfig attribute) IMAGE_INTENSITIES (direct.data.datasets.SheppLoganDataset attribute) image_model_architecture (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_mwcnn_batchnorm (direct.nn.kikinet.config.KIKINetConfig attribute) image_mwcnn_bias (direct.nn.kikinet.config.KIKINetConfig attribute) image_mwcnn_hidden_channels (direct.nn.kikinet.config.KIKINetConfig attribute) image_mwcnn_num_scales (direct.nn.kikinet.config.KIKINetConfig attribute) image_no_parameter_sharing (direct.nn.iterdualnet.config.IterDualNetConfig attribute) image_normunet (direct.nn.iterdualnet.config.IterDualNetConfig attribute) image_recon_type (direct.data.datasets_config.TransformsConfig attribute) image_resnet_batchnorm (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_resnet_hidden_channels (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_resnet_num_blocks (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_resnet_scale (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_unet_dropout (direct.nn.iterdualnet.config.IterDualNetConfig attribute) (direct.nn.jointicnet.config.JointICNetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_unet_dropout_probability (direct.nn.kikinet.config.KIKINetConfig attribute) image_unet_num_filters (direct.nn.iterdualnet.config.IterDualNetConfig attribute) (direct.nn.jointicnet.config.JointICNetConfig attribute) (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) image_unet_num_pool_layers (direct.nn.iterdualnet.config.IterDualNetConfig attribute) (direct.nn.jointicnet.config.JointICNetConfig attribute) (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) in_channels (direct.nn.resnet.config.ResNetConfig attribute) (direct.nn.unet.config.NormUnetModel2dConfig attribute) (direct.nn.unet.config.UnetModel2dConfig attribute) (direct.nn.unet.config.UnetModel3dConfig attribute) IndRNNCell (class in direct.nn.cirim.cirim) inference (direct.config.defaults.DefaultConfig attribute) inference_on_environment() (in module direct.inference) InferenceConfig (class in direct.config.defaults) init_z() (direct.nn.conjgradnet.conjgradnet.ConjGradNet static method) initialize_models_from_config() (in module direct.environment) initializer_activation (direct.nn.vsharp.config.VSharpNet3DConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) initializer_channels (direct.nn.recurrentvarnet.config.RecurrentVarNetConfig attribute) (direct.nn.rim.config.RIMConfig attribute) (direct.nn.vsharp.config.VSharpNet3DConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) initializer_dilations (direct.nn.recurrentvarnet.config.RecurrentVarNetConfig attribute) (direct.nn.rim.config.RIMConfig attribute) (direct.nn.vsharp.config.VSharpNet3DConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) initializer_initialization (direct.nn.recurrentvarnet.config.RecurrentVarNetConfig attribute) initializer_multiscale (direct.nn.recurrentvarnet.config.RecurrentVarNetConfig attribute) (direct.nn.rim.config.RIMConfig attribute) (direct.nn.vsharp.config.VSharpNet3DConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) InitType (class in direct.nn.types) INPUT_ (direct.ssl.ssl.SSLTransformMaskPrefixes attribute) INPUT_IMAGE (direct.nn.types.InitType attribute) input_image_key (direct.data.datasets_config.H5SliceConfig attribute) input_kspace_key (direct.data.datasets_config.H5SliceConfig attribute) instance_norm (direct.nn.rim.config.RIMConfig attribute) integerize_seed() (in module direct.common.subsample) IntegerListOrTupleString (class in direct.types) IntegerListOrTupleStringMeta (class in direct.types) inverted_residual_setting (direct.nn.mobilenet.config.MobileNetV2Config attribute) is_complex_data() (in module direct.utils) is_file() (in module direct.cli.utils) is_main_process() (in module direct.utils.communication) is_power_of_two() (in module direct.utils) ItemNotFoundException iter (direct.utils.events.EventStorage property) IterDualNet (class in direct.nn.iterdualnet.iterdualnet) IterDualNetConfig (class in direct.nn.iterdualnet.config) IterDualNetEngine (class in direct.nn.iterdualnet.iterdualnet_engine) IWT (class in direct.nn.mwcnn.mwcnn) J JointICNet (class in direct.nn.jointicnet.jointicnet) JointICNetConfig (class in direct.nn.jointicnet.config) JointICNetEngine (class in direct.nn.jointicnet.jointicnet_engine) JSONWriter (class in direct.utils.events) JSSLMRIModelEngine (class in direct.nn.ssl.mri_models) K KIKINet (class in direct.nn.kikinet.kikinet) KIKINetConfig (class in direct.nn.kikinet.config) KIKINetEngine (class in direct.nn.kikinet.kikinet_engine) KSPACE (direct.types.KspaceKey attribute) (direct.types.TransformKey attribute) kspace_context (direct.data.datasets_config.CMRxReconConfig attribute) (direct.data.datasets_config.H5SliceConfig attribute) kspace_conv_activation (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_conv_batchnorm (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_conv_hidden_channels (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_conv_n_convs (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_correction() (direct.nn.crossdomain.crossdomain.CrossDomainNetwork method) kspace_didn_hidden_channels (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_didn_num_convs_recon (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_didn_num_dubs (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_key (direct.data.datasets_config.CMRxReconConfig attribute) KSPACE_L1_LOSS (direct.nn.types.LossFunType attribute) KSPACE_L2_LOSS (direct.nn.types.LossFunType attribute) kspace_model_architecture (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) KSPACE_NMAE_LOSS (direct.nn.types.LossFunType attribute) KSPACE_NMSE_LOSS (direct.nn.types.LossFunType attribute) kspace_no_parameter_sharing (direct.nn.iterdualnet.config.IterDualNetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_normunet (direct.nn.iterdualnet.config.IterDualNetConfig attribute) KSPACE_NRMSE_LOSS (direct.nn.types.LossFunType attribute) kspace_resnet_batchnorm (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_resnet_hidden_channels (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_resnet_num_blocks (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_resnet_scale (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_unet_dropout (direct.nn.iterdualnet.config.IterDualNetConfig attribute) (direct.nn.jointicnet.config.JointICNetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_unet_dropout_probability (direct.nn.kikinet.config.KIKINetConfig attribute) kspace_unet_num_filters (direct.nn.iterdualnet.config.IterDualNetConfig attribute) (direct.nn.jointicnet.config.JointICNetConfig attribute) (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) kspace_unet_num_pool_layers (direct.nn.iterdualnet.config.IterDualNetConfig attribute) (direct.nn.jointicnet.config.JointICNetConfig attribute) (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) KspaceKey (class in direct.types) KtBaseMaskFunc (class in direct.common.subsample) KtGaussian1DMaskFunc (class in direct.common.subsample) KtRadialMaskFunc (class in direct.common.subsample) KtUniformMaskFunc (class in direct.common.subsample) L L1_LOSS (direct.nn.types.LossFunType attribute) L2_LOSS (direct.nn.types.LossFunType attribute) LagrangeMultipliersInitializer (class in direct.nn.vsharp.vsharp) LagrangeMultipliersInitializer3D (class in direct.nn.vsharp.vsharp) latest() (direct.utils.events.EventStorage method) (direct.utils.events.HistoryBuffer method) latest_with_smoothing_hint() (direct.utils.events.EventStorage method) launch() (in module direct.launch) launch_distributed() (in module direct.launch) LEAKY_RELU (direct.nn.types.ActivationType attribute) learned_initializer (direct.nn.recurrentvarnet.config.RecurrentVarNetConfig attribute) (direct.nn.rim.config.RIMConfig attribute) length (direct.nn.rim.config.RIMConfig attribute) linear_indices_to_2d_coordinates() (direct.common.subsample.KtBaseMaskFunc static method) load() (direct.checkpointer.Checkpointer method) load_dataset_config() (in module direct.environment) load_from_path() (direct.checkpointer.Checkpointer method) load_model_config_from_name() (in module direct.environment) load_model_from_name() (in module direct.environment) load_models_from_file() (direct.checkpointer.Checkpointer method) load_models_into_environment_config() (in module direct.environment) log_as_image (direct.config.defaults.LoggingConfig attribute) log_first_training_example_and_model() (direct.engine.Engine method) (direct.nn.ssl.mri_models.SSLMRIModelEngine method) log_process() (direct.engine.Engine method) logging (direct.config.defaults.DefaultConfig attribute) LoggingConfig (class in direct.config.defaults) loss (direct.config.defaults.TrainingConfig attribute) LossConfig (class in direct.config.defaults) losses (direct.config.defaults.LossConfig attribute) LossFunType (class in direct.nn.types) LPDNet (class in direct.nn.lpd.lpd) LPDNetConfig (class in direct.nn.lpd.config) LPDNetEngine (class in direct.nn.lpd.lpd_engine) lr (direct.config.defaults.TrainingConfig attribute) lr_gamma (direct.config.defaults.TrainingConfig attribute) lr_step_size (direct.config.defaults.TrainingConfig attribute) lr_warmup_iter (direct.config.defaults.TrainingConfig attribute) LRScheduler (class in direct.data.lr_scheduler) M MagicMaskFunc (class in direct.common.subsample) main() (in module direct.cli) make_blobs() (direct.data.fake.FakeMRIData method) mask_func() (direct.common.subsample.BaseMaskFunc method) (direct.common.subsample.CalgaryCampinasMaskFunc method) (direct.common.subsample.CIRCUSMaskFunc method) (direct.common.subsample.EquispacedMaskFunc method) (direct.common.subsample.Gaussian1DMaskFunc method) (direct.common.subsample.Gaussian2DMaskFunc method) (direct.common.subsample.KtGaussian1DMaskFunc method) (direct.common.subsample.KtRadialMaskFunc method) (direct.common.subsample.KtUniformMaskFunc method) (direct.common.subsample.MagicMaskFunc method) (direct.common.subsample.RandomMaskFunc method) (direct.common.subsample.VariableDensityPoissonMaskFunc method) MASK_MD5S (direct.common.subsample.CalgaryCampinasMaskFunc attribute) mask_split_acs_region (direct.data.datasets_config.TransformsConfig attribute) mask_split_gaussian_std (direct.data.datasets_config.TransformsConfig attribute) mask_split_half_direction (direct.data.datasets_config.TransformsConfig attribute) mask_split_keep_acs (direct.data.datasets_config.TransformsConfig attribute) mask_split_ratio (direct.data.datasets_config.TransformsConfig attribute) mask_split_type (direct.data.datasets_config.TransformsConfig attribute) MASKED_KSPACE (direct.types.KspaceKey attribute) (direct.types.TransformKey attribute) MaskFuncMode (class in direct.common.subsample) (class in direct.types) masking (direct.data.datasets_config.TransformsConfig attribute) MaskingConfig (class in direct.common.subsample_config) MaskSplitterType (class in direct.ssl.ssl) MaximumEigenvaluePowerMethod (class in direct.algorithms.optimization) median() (direct.utils.events.HistoryBuffer method) merge_list_of_dicts() (in module direct.utils) metrics (direct.config.defaults.TrainingConfig attribute) (direct.config.defaults.ValidationConfig attribute) MobileNetV2 (class in direct.nn.mobilenet.mobilenet) MobileNetV2Config (class in direct.nn.mobilenet.config) mode (direct.common.subsample_config.MaskingConfig attribute) model (direct.config.defaults.DefaultConfig attribute) model_checkpoint (direct.config.defaults.TrainingConfig attribute) model_name (direct.config.defaults.ModelConfig attribute) ModelConfig (class in direct.config.defaults) ModelName (class in direct.nn.types) models_to_device() (direct.engine.Engine method) models_training_mode() (direct.engine.Engine method) models_validation_mode() (direct.engine.Engine method) module direct direct.algorithms direct.algorithms.mri_algorithms direct.algorithms.optimization direct.checkpointer direct.cli direct.cli.predict direct.cli.train direct.cli.upload direct.cli.utils direct.common direct.common.subsample direct.common.subsample_config direct.config direct.config.defaults direct.constants direct.data direct.data.bbox direct.data.datasets direct.data.datasets_config direct.data.fake direct.data.h5_data direct.data.lr_scheduler direct.data.mri_transforms direct.data.samplers direct.data.sens direct.data.transforms direct.engine direct.environment direct.exceptions direct.functionals direct.functionals.challenges direct.functionals.grad direct.functionals.hfen direct.functionals.nmae direct.functionals.nmse direct.functionals.psnr direct.functionals.snr direct.functionals.ssim direct.inference direct.launch direct.nn direct.nn.cirim direct.nn.cirim.cirim direct.nn.cirim.cirim_engine direct.nn.cirim.config direct.nn.conjgradnet direct.nn.conjgradnet.config direct.nn.conjgradnet.conjgrad direct.nn.conjgradnet.conjgradnet direct.nn.conjgradnet.conjgradnet_engine direct.nn.conv direct.nn.conv.conv direct.nn.crossdomain direct.nn.crossdomain.crossdomain direct.nn.crossdomain.multicoil direct.nn.didn direct.nn.didn.didn direct.nn.get_nn_model_config direct.nn.iterdualnet direct.nn.iterdualnet.config direct.nn.iterdualnet.iterdualnet direct.nn.iterdualnet.iterdualnet_engine direct.nn.jointicnet direct.nn.jointicnet.config direct.nn.jointicnet.jointicnet direct.nn.jointicnet.jointicnet_engine direct.nn.kikinet direct.nn.kikinet.config direct.nn.kikinet.kikinet direct.nn.kikinet.kikinet_engine direct.nn.lpd direct.nn.lpd.config direct.nn.lpd.lpd direct.nn.lpd.lpd_engine direct.nn.mobilenet direct.nn.mobilenet.config direct.nn.mobilenet.mobilenet direct.nn.mri_models direct.nn.multidomainnet direct.nn.multidomainnet.config direct.nn.multidomainnet.multidomain direct.nn.multidomainnet.multidomainnet direct.nn.multidomainnet.multidomainnet_engine direct.nn.mwcnn direct.nn.mwcnn.mwcnn direct.nn.recurrent direct.nn.recurrent.recurrent direct.nn.recurrentvarnet direct.nn.recurrentvarnet.config direct.nn.recurrentvarnet.recurrentvarnet direct.nn.recurrentvarnet.recurrentvarnet_engine direct.nn.resnet direct.nn.resnet.config direct.nn.resnet.resnet direct.nn.rim direct.nn.rim.config direct.nn.rim.rim direct.nn.rim.rim_engine direct.nn.ssl direct.nn.ssl.mri_models direct.nn.types direct.nn.unet direct.nn.unet.config direct.nn.unet.unet_2d direct.nn.unet.unet_3d direct.nn.unet.unet_engine direct.nn.varnet direct.nn.varnet.config direct.nn.varnet.varnet direct.nn.varnet.varnet_engine direct.nn.varsplitnet direct.nn.varsplitnet.config direct.nn.varsplitnet.varsplitnet direct.nn.varsplitnet.varsplitnet_engine direct.nn.vsharp direct.nn.vsharp.config direct.nn.vsharp.vsharp direct.nn.vsharp.vsharp_engine direct.nn.xpdnet direct.nn.xpdnet.config direct.nn.xpdnet.xpdnet direct.nn.xpdnet.xpdnet_engine direct.predict direct.ssl direct.ssl.mask_fillers direct.ssl.ssl direct.train direct.types direct.utils direct.utils.asserts direct.utils.bbox direct.utils.communication direct.utils.dataset direct.utils.events direct.utils.imports direct.utils.io direct.utils.logging direct.utils.models direct.utils.writers ModuleWrapper (class in direct.data.mri_transforms) ModuleWrapper.SubWrapper (class in direct.data.mri_transforms) modulus() (in module direct.data.transforms) modulus_if_complex() (in module direct.data.transforms) MRILogLikelihood (class in direct.nn.rim.rim) MRIModelEngine (class in direct.nn.mri_models) MRIVarSplitNet (class in direct.nn.varsplitnet.varsplitnet) MRIVarSplitNetConfig (class in direct.nn.varsplitnet.config) MRIVarSplitNetEngine (class in direct.nn.varsplitnet.varsplitnet_engine) MultiCoil (class in direct.nn.crossdomain.multicoil) MultiDomainConv2d (class in direct.nn.multidomainnet.multidomain) MultiDomainConvBlock (class in direct.nn.multidomainnet.multidomain) MultiDomainConvTranspose2d (class in direct.nn.multidomainnet.multidomain) MultiDomainNet (class in direct.nn.multidomainnet.multidomainnet) MultiDomainNetConfig (class in direct.nn.multidomainnet.config) MultiDomainNetEngine (class in direct.nn.multidomainnet.multidomainnet_engine) MultiDomainUnet2d (class in direct.nn.multidomainnet.multidomain) multiplier (direct.config.defaults.FunctionConfig attribute) multiply_function() (in module direct.utils) MULTISLICE (direct.common.subsample.MaskFuncMode attribute) (direct.types.MaskFuncMode attribute) MWCNN (class in direct.nn.mwcnn.mwcnn) mwcnn_batchnorm (direct.nn.xpdnet.config.XPDNetConfig attribute) mwcnn_bias (direct.nn.xpdnet.config.XPDNetConfig attribute) mwcnn_hidden_channels (direct.nn.xpdnet.config.XPDNetConfig attribute) mwcnn_num_scales (direct.nn.xpdnet.config.XPDNetConfig attribute) N name (direct.common.subsample_config.MaskingConfig attribute) (direct.data.datasets_config.DatasetConfig attribute) name_scope() (direct.utils.events.EventStorage method) ndim (direct.engine.DataDimensionality property) NEAREST (direct.data.mri_transforms.RescaleMode attribute) NEAREST_EXACT (direct.data.mri_transforms.RescaleMode attribute) NMAE_LOSS (direct.nn.types.LossFunType attribute) NMAELoss (class in direct.functionals) (class in direct.functionals.nmae) NMSE_LOSS (direct.nn.types.LossFunType attribute) NMSELoss (class in direct.functionals) (class in direct.functionals.nmse) no_parameter_sharing (direct.nn.cirim.config.CIRIMConfig attribute) (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) (direct.nn.recurrentvarnet.config.RecurrentVarNetConfig attribute) (direct.nn.rim.config.RIMConfig attribute) (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) (direct.nn.vsharp.config.VSharpNet3DConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) noise_matrix_scaling (direct.config.defaults.PhysicsConfig attribute) norm() (direct.nn.recurrent.recurrent.NormConv2dGRU static method) (direct.nn.unet.unet_2d.NormUnetModel2d static method) (direct.nn.unet.unet_3d.NormUnetModel3d static method) norm_groups (direct.nn.unet.config.NormUnetModel2dConfig attribute) norm_layer (direct.nn.mobilenet.config.MobileNetV2Config attribute) normalization (direct.data.datasets_config.TransformsConfig attribute) NormalizationTransformConfig (class in direct.data.datasets_config) normalize (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) normalize_image() (in module direct.utils) normalized (direct.nn.recurrentvarnet.config.RecurrentVarNetConfig attribute) (direct.nn.rim.config.RIMConfig attribute) (direct.nn.unet.config.Unet2dConfig attribute) NormalizeModule (class in direct.data.mri_transforms) NormConv2dGRU (class in direct.nn.recurrent.recurrent) NORMUNET (direct.nn.types.ModelName attribute) NormUnetModel2d (class in direct.nn.unet.unet_2d) NormUnetModel2dConfig (class in direct.nn.unet.config) NormUnetModel3d (class in direct.nn.unet.unet_3d) NRMSE_LOSS (direct.nn.types.LossFunType attribute) NRMSELoss (class in direct.functionals) (class in direct.functionals.nmse) NUM_ACS_LINES (direct.data.datasets.CMRxReconDataset attribute) num_blocks (direct.nn.resnet.config.ResNetConfig attribute) num_cascades (direct.nn.cirim.config.CIRIMConfig attribute) num_channels (direct.nn.mobilenet.config.MobileNetV2Config attribute) num_classes (direct.nn.mobilenet.config.MobileNetV2Config attribute) num_coils (direct.data.datasets_config.SheppLoganDatasetConfig attribute) num_dual (direct.nn.lpd.config.LPDNetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) num_filters (direct.nn.multidomainnet.config.MultiDomainNetConfig attribute) (direct.nn.unet.config.NormUnetModel2dConfig attribute) (direct.nn.unet.config.Unet2dConfig attribute) (direct.nn.unet.config.UnetModel2dConfig attribute) (direct.nn.unet.config.UnetModel3dConfig attribute) num_images (direct.config.defaults.TensorboardConfig attribute) num_iter (direct.nn.iterdualnet.config.IterDualNetConfig attribute) (direct.nn.jointicnet.config.JointICNetConfig attribute) (direct.nn.kikinet.config.KIKINetConfig attribute) (direct.nn.lpd.config.LPDNetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) num_iterations (direct.config.defaults.TrainingConfig attribute) num_layers (direct.nn.varnet.config.EndToEndVarNetConfig attribute) num_pool_layers (direct.nn.multidomainnet.config.MultiDomainNetConfig attribute) (direct.nn.unet.config.NormUnetModel2dConfig attribute) (direct.nn.unet.config.Unet2dConfig attribute) (direct.nn.unet.config.UnetModel2dConfig attribute) (direct.nn.unet.config.UnetModel3dConfig attribute) num_primal (direct.nn.lpd.config.LPDNetConfig attribute) (direct.nn.xpdnet.config.XPDNetConfig attribute) num_steps (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) (direct.nn.recurrentvarnet.config.RecurrentVarNetConfig attribute) (direct.nn.vsharp.config.VSharpNet3DConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) num_steps_dc (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) num_steps_dc_gd (direct.nn.vsharp.config.VSharpNet3DConfig attribute) (direct.nn.vsharp.config.VSharpNetConfig attribute) num_steps_reg (direct.nn.varsplitnet.config.MRIVarSplitNetConfig attribute) O optimizer (direct.config.defaults.TrainingConfig attribute) orthotogonalize_weights() (direct.nn.cirim.cirim.IndRNNCell static method) out_channels (direct.nn.resnet.config.ResNetConfig attribute) (direct.nn.unet.config.NormUnetModel2dConfig attribute) (direct.nn.unet.config.UnetModel2dConfig attribute) (direct.nn.unet.config.UnetModel3dConfig attribute) output_image (direct.engine.DoIterationOutput attribute) P pad (direct.data.datasets_config.AugmentationTransformConfig attribute) pad() (direct.nn.didn.didn.DUB static method) (direct.nn.mwcnn.mwcnn.MWCNN static method) (direct.nn.unet.unet_2d.NormUnetModel2d static method) (direct.nn.unet.unet_3d.NormUnetModel3d static method) pad_coils (direct.data.datasets_config.TransformsConfig attribute) pad_tensor() (in module direct.data.transforms) pad_to_pow_of_2() (in module direct.nn.unet.unet_3d) PadCoilDimensionModule (class in direct.data.mri_transforms) padding_eps (direct.data.datasets_config.TransformsConfig attribute) PadKspace (class in direct.data.mri_transforms) parse_filenames_data() (direct.data.datasets.CMRxReconDataset method) (direct.data.datasets.FakeMRIBlobsDataset method) (direct.data.h5_data.H5SliceData method) parse_noise_dict() (in module direct.train) pass_attrs (direct.data.datasets_config.FakeMRIBlobsConfig attribute) (direct.data.datasets_config.FastMRIConfig attribute) pass_mask (direct.data.datasets_config.H5SliceConfig attribute) physics (direct.config.defaults.DefaultConfig attribute) PhysicsConfig (class in direct.config.defaults) poisson() (direct.common.subsample.VariableDensityPoissonMaskFunc method) predict() (direct.engine.Engine method) predict_from_argparse() (in module direct.predict) prefix_dict_keys() (in module direct.utils) PRELU (direct.nn.types.ActivationType attribute) primal_model_architecture (direct.nn.lpd.config.LPDNetConfig attribute) primal_mwcnn_batchnorm (direct.nn.lpd.config.LPDNetConfig attribute) primal_mwcnn_bias (direct.nn.lpd.config.LPDNetConfig attribute) primal_mwcnn_hidden_channels (direct.nn.lpd.config.LPDNetConfig attribute) primal_mwcnn_num_scales (direct.nn.lpd.config.LPDNetConfig attribute) primal_unet_dropout_probability (direct.nn.lpd.config.LPDNetConfig attribute) primal_unet_num_filters (direct.nn.lpd.config.LPDNetConfig attribute) primal_unet_num_pool_layers (direct.nn.lpd.config.LPDNetConfig attribute) PrimalNet (class in direct.nn.lpd.lpd) process_slices_for_visualization() (direct.engine.Engine method) ProcessKilledException PRP (direct.nn.conjgradnet.conjgrad.CGUpdateType attribute) PSNR_LOSS (direct.nn.types.LossFunType attribute) PSNRLoss (class in direct.functionals) (class in direct.functionals.psnr) R RadialMaskFunc (class in direct.common.subsample) RANDOM (direct.data.mri_transforms.RandomFlipType attribute) random_augmentations (direct.data.datasets_config.TransformsConfig attribute) random_flip_probability (direct.data.datasets_config.RandomAugmentationTransformsConfig attribute) random_flip_type (direct.data.datasets_config.RandomAugmentationTransformsConfig attribute) random_reverse_probability (direct.data.datasets_config.RandomAugmentationTransformsConfig attribute) random_rotation_degrees (direct.data.datasets_config.RandomAugmentationTransformsConfig attribute) random_rotation_probability (direct.data.datasets_config.RandomAugmentationTransformsConfig attribute) RandomAugmentationTransformsConfig (class in direct.data.datasets_config) RandomFlip (class in direct.data.mri_transforms) RandomFlipType (class in direct.data.mri_transforms) RandomMaskFunc (class in direct.common.subsample) RandomReverse (class in direct.data.mri_transforms) RandomRotation (class in direct.data.mri_transforms) read_json() (in module direct.utils.io) read_list() (in module direct.utils.io) read_text_from_url() (in module direct.utils.io) ReconBlock (class in direct.nn.didn.didn) reconstruct_volumes() (direct.engine.Engine method) (direct.nn.mri_models.MRIModelEngine method) ReconstructionType (class in direct.data.mri_transforms) recurrent_hidden_channels (direct.nn.cirim.config.CIRIMConfig attribute) (direct.nn.recurrentvarnet.config.RecurrentVarNetConfig attribute) recurrent_num_layers (direct.nn.recurrentvarnet.config.RecurrentVarNetConfig attribute) RecurrentInit (class in direct.nn.recurrentvarnet.recurrentvarnet) RecurrentVarNet (class in direct.nn.recurrentvarnet.recurrentvarnet) RecurrentVarNetBlock (class in direct.nn.recurrentvarnet.recurrentvarnet) RecurrentVarNetConfig (class in direct.nn.recurrentvarnet.config) RecurrentVarNetEngine (class in direct.nn.recurrentvarnet.recurrentvarnet_engine) reduce_list_of_dicts() (in module direct.utils) reduce_operator() (in module direct.data.transforms) reduce_tensor_dict() (in module direct.utils.communication) regex_filter (direct.data.datasets_config.CMRxReconConfig attribute) (direct.data.datasets_config.H5SliceConfig attribute) register_parser() (in module direct.cli.predict) (in module direct.cli.train) (in module direct.cli.upload) regularizer_dropout (direct.nn.varnet.config.EndToEndVarNetConfig attribute) regularizer_num_filters (direct.nn.varnet.config.EndToEndVarNetConfig attribute) regularizer_num_pull_layers (direct.nn.varnet.config.EndToEndVarNetConfig attribute) regularizers (direct.config.defaults.TrainingConfig attribute) (direct.config.defaults.ValidationConfig attribute) RELU (direct.nn.types.ActivationType attribute) remove_keys() (in module direct.utils) RenameKeysModule (class in direct.data.mri_transforms) replication_padding (direct.nn.rim.config.RIMConfig attribute) rescale (direct.data.datasets_config.AugmentationTransformConfig attribute) rescale_2d_if_3d (direct.data.datasets_config.AugmentationTransformConfig attribute) rescale_mode (direct.data.datasets_config.AugmentationTransformConfig attribute) RescaleKspace (class in direct.data.mri_transforms) RescaleMode (class in direct.data.mri_transforms) resenet_batchnorm (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) resenet_scale (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) reset_parameters() (direct.nn.cirim.cirim.ConvNonlinear method) (direct.nn.cirim.cirim.IndRNNCell method) ResNet (class in direct.nn.resnet.resnet) RESNET (direct.nn.types.ModelName attribute) resnet_hidden_channels (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) resnet_num_blocks (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) ResNetBlock (class in direct.nn.resnet.resnet) ResNetConfig (class in direct.nn.resnet.config) resolve_duplicates_on_kt_grid() (direct.common.subsample.KtBaseMaskFunc static method) RIM (class in direct.nn.rim.rim) RIMBlock (class in direct.nn.cirim.cirim) RIMConfig (class in direct.nn.rim.config) RIMEngine (class in direct.nn.rim.rim_engine) RIMInit (class in direct.nn.rim.rim) roll() (in module direct.data.transforms) roll_one_dim() (in module direct.data.transforms) root_sum_of_squares() (in module direct.data.fake) (in module direct.data.transforms) round_nearest (direct.nn.mobilenet.config.MobileNetV2Config attribute) RSS (direct.data.mri_transforms.ReconstructionType attribute) RSS_ESTIMATE (direct.data.mri_transforms.SensitivityMapType attribute) S safe_divide() (in module direct.data.transforms) sample_image() (direct.data.datasets.SheppLoganDataset method) SAMPLING_MASK (direct.types.TransformKey attribute) save() (direct.checkpointer.Checkpointer method) scale (direct.nn.resnet.config.ResNetConfig attribute) scale_data() (in module direct.data.fake) scale_loglikelihood (direct.nn.rim.config.RIMConfig attribute) scale_percentile (direct.data.datasets_config.NormalizationTransformConfig attribute) SCALING_FACTOR (direct.types.TransformKey attribute) scaling_key (direct.data.datasets_config.NormalizationTransformConfig attribute) seed (direct.data.datasets_config.SheppLoganDatasetConfig attribute) sens_unet_dropout (direct.nn.jointicnet.config.JointICNetConfig attribute) sens_unet_num_filters (direct.nn.jointicnet.config.JointICNetConfig attribute) sens_unet_num_pool_layers (direct.nn.jointicnet.config.JointICNetConfig attribute) SENSE (direct.data.mri_transforms.ReconstructionType attribute) (direct.nn.types.InitType attribute) SENSE_MOD (direct.data.mri_transforms.ReconstructionType attribute) sensitivity_map (direct.engine.DoIterationOutput attribute) SENSITIVITY_MAP (direct.types.TransformKey attribute) sensitivity_map_estimation (direct.data.datasets_config.TransformsConfig attribute) sensitivity_maps_espirit_crop (direct.data.datasets_config.SensitivityMapEstimationTransformConfig attribute) sensitivity_maps_espirit_kernel_size (direct.data.datasets_config.SensitivityMapEstimationTransformConfig attribute) sensitivity_maps_espirit_max_iters (direct.data.datasets_config.SensitivityMapEstimationTransformConfig attribute) sensitivity_maps_espirit_threshold (direct.data.datasets_config.SensitivityMapEstimationTransformConfig attribute) sensitivity_maps_gaussian (direct.data.datasets_config.SensitivityMapEstimationTransformConfig attribute) sensitivity_maps_type (direct.data.datasets_config.SensitivityMapEstimationTransformConfig attribute) SensitivityMapEstimationTransformConfig (class in direct.data.datasets_config) SensitivityMapType (class in direct.data.mri_transforms) set_all_seeds() (in module direct.utils) set_attrs() (direct.data.fake.FakeMRIData method) setup() (in module direct.utils.logging) setup_common_environment() (in module direct.environment) setup_engine() (in module direct.environment) setup_inference_environment() (in module direct.environment) setup_inference_save_to_h5() (in module direct.inference) setup_logging() (in module direct.environment) setup_testing_environment() (in module direct.environment) setup_train() (in module direct.train) setup_training_environment() (in module direct.environment) shape (direct.data.datasets_config.SheppLoganDatasetConfig attribute) shared_random_seed() (in module direct.utils.communication) SheppLoganDataset (class in direct.data.datasets) SheppLoganDatasetConfig (class in direct.data.datasets_config) SheppLoganProtonConfig (class in direct.data.datasets_config) SheppLoganProtonDataset (class in direct.data.datasets) SheppLoganT1Config (class in direct.data.datasets_config) SheppLoganT1Dataset (class in direct.data.datasets) SheppLoganT2Config (class in direct.data.datasets_config) SheppLoganT2Dataset (class in direct.data.datasets) simulate_sensitivity_maps() (in module direct.data.sens) skip_connection (direct.nn.unet.config.Unet2dConfig attribute) smoothing_hints() (direct.utils.events.EventStorage method) SNR_LOSS (direct.nn.types.LossFunType attribute) snr_metric() (in module direct.functionals) (in module direct.functionals.snr) SNRLoss (class in direct.functionals) (class in direct.functionals.snr) SobelGradL1Loss (class in direct.functionals) (class in direct.functionals.grad) SobelGradL2Loss (class in direct.functionals) (class in direct.functionals.grad) SpatialDims (class in direct.utils) SpiralMaskFunc (class in direct.common.subsample) split_method() (direct.ssl.ssl.GaussianMaskSplitterModule method) (direct.ssl.ssl.HalfMaskSplitterModule method) (direct.ssl.ssl.UniformMaskSplitterModule method) SSIM3DLoss (class in direct.functionals) (class in direct.functionals.ssim) SSIM_3D_LOSS (direct.nn.types.LossFunType attribute) SSIM_LOSS (direct.nn.types.LossFunType attribute) SSIMLoss (class in direct.functionals) (class in direct.functionals.ssim) SSL_SSDU (direct.data.mri_transforms.TransformsType attribute) SSLMRIModelEngine (class in direct.nn.ssl.mri_models) SSLTransformMaskPrefixes (class in direct.ssl.ssl) standardization (direct.nn.multidomainnet.config.MultiDomainNetConfig attribute) StandardizationLayer (class in direct.nn.multidomainnet.multidomainnet) state_dict() (direct.data.lr_scheduler.LRScheduler method) STATIC (direct.common.subsample.MaskFuncMode attribute) (direct.types.MaskFuncMode attribute) step() (direct.utils.events.EventStorage method) steps (direct.nn.rim.config.RIMConfig attribute) str_to_class() (in module direct.utils) Subpixel (class in direct.nn.didn.didn) SUPERVISED (direct.data.mri_transforms.TransformsType attribute) swa_start_iter (direct.config.defaults.TrainingConfig attribute) synchronize() (in module direct.utils.communication) T T2_star (direct.data.datasets_config.SheppLoganT2Config attribute) TARGET (direct.types.TransformKey attribute) TARGET_ (direct.ssl.ssl.SSLTransformMaskPrefixes attribute) tensor_to_complex_numpy() (in module direct.data.transforms) tensorboard (direct.config.defaults.LoggingConfig attribute) TensorboardConfig (class in direct.config.defaults) TensorboardWriter (class in direct.utils.events) text_description (direct.data.datasets_config.DatasetConfig attribute) THREE_D (direct.utils.SpatialDims attribute) time_steps (direct.nn.cirim.config.CIRIMConfig attribute) to_tensor() (in module direct.data.transforms) ToTensor (class in direct.data.mri_transforms) train() (direct.engine.Engine method) train_from_argparse() (in module direct.train) training (direct.algorithms.mri_algorithms.EspiritCalibration attribute) (direct.config.defaults.DefaultConfig attribute) (direct.data.mri_transforms.AddBooleanKeysModule attribute) (direct.data.mri_transforms.ApplyMaskModule attribute) (direct.data.mri_transforms.CompressCoilModule attribute) (direct.data.mri_transforms.ComputeImageModule attribute) (direct.data.mri_transforms.ComputeScalingFactorModule attribute) (direct.data.mri_transforms.DeleteKeysModule attribute) (direct.data.mri_transforms.EstimateSensitivityMapModule attribute) (direct.data.mri_transforms.NormalizeModule attribute) (direct.data.mri_transforms.PadCoilDimensionModule attribute) (direct.data.mri_transforms.RenameKeysModule attribute) (direct.data.mri_transforms.WhitenDataModule attribute) (direct.functionals.grad.SobelGradL1Loss attribute) (direct.functionals.grad.SobelGradL2Loss attribute) (direct.functionals.hfen.HFENL1Loss attribute) (direct.functionals.hfen.HFENL2Loss attribute) (direct.functionals.hfen.HFENLoss attribute) (direct.functionals.HFENL1Loss attribute) (direct.functionals.HFENL2Loss attribute) (direct.functionals.HFENLoss attribute) (direct.functionals.nmae.NMAELoss attribute) (direct.functionals.NMAELoss attribute) (direct.functionals.nmse.NMSELoss attribute) (direct.functionals.nmse.NRMSELoss attribute) (direct.functionals.NMSELoss attribute) (direct.functionals.NRMSELoss attribute) (direct.functionals.psnr.PSNRLoss attribute) (direct.functionals.PSNRLoss attribute) (direct.functionals.snr.SNRLoss attribute) (direct.functionals.SNRLoss attribute) (direct.functionals.SobelGradL1Loss attribute) (direct.functionals.SobelGradL2Loss attribute) (direct.functionals.ssim.SSIM3DLoss attribute) (direct.functionals.ssim.SSIMLoss attribute) (direct.functionals.SSIM3DLoss attribute) (direct.functionals.SSIMLoss attribute) (direct.nn.cirim.cirim.CIRIM attribute) (direct.nn.cirim.cirim.ConvNonlinear attribute) (direct.nn.cirim.cirim.ConvRNNStack attribute) (direct.nn.cirim.cirim.IndRNNCell attribute) (direct.nn.cirim.cirim.RIMBlock attribute) (direct.nn.conjgradnet.conjgrad.ConjGrad attribute) (direct.nn.conjgradnet.conjgradnet.ConjGradNet attribute) (direct.nn.conv.conv.Conv2d attribute) (direct.nn.crossdomain.crossdomain.CrossDomainNetwork attribute) (direct.nn.crossdomain.multicoil.MultiCoil attribute) (direct.nn.didn.didn.DIDN attribute) (direct.nn.didn.didn.DUB attribute) (direct.nn.didn.didn.ReconBlock attribute) (direct.nn.didn.didn.Subpixel attribute) (direct.nn.iterdualnet.iterdualnet.IterDualNet attribute) (direct.nn.jointicnet.jointicnet.JointICNet attribute) (direct.nn.kikinet.kikinet.KIKINet attribute) (direct.nn.lpd.lpd.DualNet attribute) (direct.nn.lpd.lpd.LPDNet attribute) (direct.nn.lpd.lpd.PrimalNet attribute) (direct.nn.mobilenet.mobilenet.MobileNetV2 attribute) (direct.nn.multidomainnet.multidomain.MultiDomainConv2d attribute) (direct.nn.multidomainnet.multidomain.MultiDomainConvBlock attribute) (direct.nn.multidomainnet.multidomain.MultiDomainConvTranspose2d attribute) (direct.nn.multidomainnet.multidomain.MultiDomainUnet2d attribute) (direct.nn.multidomainnet.multidomain.TransposeMultiDomainConvBlock attribute) (direct.nn.multidomainnet.multidomainnet.MultiDomainNet attribute) (direct.nn.multidomainnet.multidomainnet.StandardizationLayer attribute) (direct.nn.mwcnn.mwcnn.ConvBlock attribute) (direct.nn.mwcnn.mwcnn.DilatedConvBlock attribute) (direct.nn.mwcnn.mwcnn.DWT attribute) (direct.nn.mwcnn.mwcnn.IWT attribute) (direct.nn.mwcnn.mwcnn.MWCNN attribute) (direct.nn.recurrent.recurrent.Conv2dGRU attribute) (direct.nn.recurrent.recurrent.NormConv2dGRU attribute) (direct.nn.recurrentvarnet.recurrentvarnet.RecurrentInit attribute) (direct.nn.recurrentvarnet.recurrentvarnet.RecurrentVarNet attribute) (direct.nn.recurrentvarnet.recurrentvarnet.RecurrentVarNetBlock attribute) (direct.nn.resnet.resnet.ResNet attribute) (direct.nn.resnet.resnet.ResNetBlock attribute) (direct.nn.rim.rim.MRILogLikelihood attribute) (direct.nn.rim.rim.RIM attribute) (direct.nn.rim.rim.RIMInit attribute) (direct.nn.unet.unet_2d.ConvBlock attribute) (direct.nn.unet.unet_2d.NormUnetModel2d attribute) (direct.nn.unet.unet_2d.TransposeConvBlock attribute) (direct.nn.unet.unet_2d.Unet2d attribute) (direct.nn.unet.unet_2d.UnetModel2d attribute) (direct.nn.unet.unet_3d.ConvBlock3D attribute) (direct.nn.unet.unet_3d.NormUnetModel3d attribute) (direct.nn.unet.unet_3d.TransposeConvBlock3D attribute) (direct.nn.unet.unet_3d.UnetModel3d attribute) (direct.nn.varnet.varnet.EndToEndVarNet attribute) (direct.nn.varnet.varnet.EndToEndVarNetBlock attribute) (direct.nn.varsplitnet.varsplitnet.MRIVarSplitNet attribute) (direct.nn.vsharp.vsharp.LagrangeMultipliersInitializer attribute) (direct.nn.vsharp.vsharp.LagrangeMultipliersInitializer3D attribute) (direct.nn.vsharp.vsharp.VSharpNet attribute) (direct.nn.vsharp.vsharp.VSharpNet3D attribute) (direct.nn.xpdnet.xpdnet.XPDNet attribute) (direct.ssl.ssl.GaussianMaskSplitterModule attribute) (direct.ssl.ssl.HalfMaskSplitterModule attribute) (direct.ssl.ssl.UniformMaskSplitterModule attribute) (direct.utils.DirectModule attribute) training_loop() (direct.engine.Engine method) TrainingConfig (class in direct.config.defaults) TrainingException TransformKey (class in direct.types) transforms (direct.data.datasets_config.DatasetConfig attribute) transforms_type (direct.data.datasets_config.TransformsConfig attribute) TransformsConfig (class in direct.data.datasets_config) TransformsType (class in direct.data.mri_transforms) TransposeConvBlock (class in direct.nn.unet.unet_2d) TransposeConvBlock3D (class in direct.nn.unet.unet_3d) TransposeMultiDomainConvBlock (class in direct.nn.multidomainnet.multidomain) TRILINEAR (direct.data.mri_transforms.RescaleMode attribute) TWO_D (direct.utils.SpatialDims attribute) U UNET (direct.nn.types.ModelName attribute) Unet2d (class in direct.nn.unet.unet_2d) Unet2dConfig (class in direct.nn.unet.config) Unet2dEngine (class in direct.nn.unet.unet_engine) Unet2dJSSLEngine (class in direct.nn.unet.unet_engine) Unet2dSSLEngine (class in direct.nn.unet.unet_engine) unet_dropout (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) (direct.nn.vsharp.config.VSharpNet3DConfig attribute) unet_norm (direct.nn.vsharp.config.VSharpNet3DConfig attribute) unet_num_filters (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) (direct.nn.vsharp.config.VSharpNet3DConfig attribute) unet_num_pool_layers (direct.nn.conjgradnet.config.ConjGradNetConfig attribute) (direct.nn.vsharp.config.VSharpNet3DConfig attribute) UnetModel2d (class in direct.nn.unet.unet_2d) UnetModel2dConfig (class in direct.nn.unet.config) UnetModel3d (class in direct.nn.unet.unet_3d) UnetModel3dConfig (class in direct.nn.unet.config) UNIFORM (direct.ssl.ssl.MaskSplitterType attribute) uniform_fill() (in module direct.ssl.mask_fillers) uniform_range (direct.common.subsample_config.MaskingConfig attribute) UniformMaskSplitterModule (class in direct.ssl.ssl) UNIT (direct.data.mri_transforms.SensitivityMapType attribute) unnorm() (direct.nn.recurrent.recurrent.NormConv2dGRU static method) (direct.nn.unet.unet_2d.NormUnetModel2d static method) (direct.nn.unet.unet_3d.NormUnetModel3d static method) unpad() (direct.nn.unet.unet_2d.NormUnetModel2d static method) (direct.nn.unet.unet_3d.NormUnetModel3d static method) update() (direct.algorithms.optimization.Algorithm method) (direct.utils.events.HistoryBuffer method) upload_from_argparse() (in module direct.cli.upload) upload_to_s3() (in module direct.utils.io) use_noise_matrix (direct.config.defaults.PhysicsConfig attribute) use_norm_unet (direct.nn.jointicnet.config.JointICNetConfig attribute) use_primal_only (direct.nn.xpdnet.config.XPDNetConfig attribute) use_seed (direct.data.datasets_config.TransformsConfig attribute) V val_accelerations (direct.common.subsample_config.MaskingConfig attribute) val_center_fractions (direct.common.subsample_config.MaskingConfig attribute) VALID_CHALLENGE_ACCELERATIONS (direct.data.datasets.CMRxReconDataset attribute) validate_model_at_interval() (direct.engine.Engine method) validation (direct.config.defaults.DefaultConfig attribute) validation_loop() (direct.engine.Engine method) validation_steps (direct.config.defaults.TrainingConfig attribute) ValidationConfig (class in direct.config.defaults) values() (direct.utils.events.HistoryBuffer method) VariableDensityPoissonMaskFunc (class in direct.common.subsample) verify_extra_h5_integrity() (direct.data.h5_data.H5SliceData static method) verify_extra_mat_integrity() (direct.data.datasets.CMRxReconDataset static method) verify_fft_dtype_possible() (in module direct.data.transforms) VERTICAL (direct.data.mri_transforms.RandomFlipType attribute) (direct.ssl.ssl.HalfSplitType attribute) view_as_complex() (in module direct.data.transforms) view_as_real() (in module direct.data.transforms) vis_data (direct.utils.events.EventStorage property) VSharpNet (class in direct.nn.vsharp.vsharp) VSharpNet3D (class in direct.nn.vsharp.vsharp) VSharpNet3DConfig (class in direct.nn.vsharp.config) VSharpNet3DEngine (class in direct.nn.vsharp.vsharp_engine) VSharpNetConfig (class in direct.nn.vsharp.config) VSharpNetEngine (class in direct.nn.vsharp.vsharp_engine) VSharpNetJSSLEngine (class in direct.nn.vsharp.vsharp_engine) VSharpNetSSLEngine (class in direct.nn.vsharp.vsharp_engine) W WarmupCosineLR (class in direct.data.lr_scheduler) WarmupMultiStepLR (class in direct.data.lr_scheduler) weight_decay (direct.config.defaults.TrainingConfig attribute) whiten_input (direct.nn.rim.config.RIMConfig attribute) WhitenDataModule (class in direct.data.mri_transforms) width_mult (direct.nn.mobilenet.config.MobileNetV2Config attribute) write() (direct.utils.events.CommonMetricPrinter method) (direct.utils.events.EventWriter method) (direct.utils.events.JSONWriter method) (direct.utils.events.TensorboardWriter method) write_json() (in module direct.utils.io) write_list() (in module direct.utils.io) write_output_to_h5() (in module direct.utils.writers) write_to_logs() (direct.engine.Engine method) write_to_logs_at_interval() (direct.engine.Engine method) X XPDNet (class in direct.nn.xpdnet.xpdnet) XPDNetConfig (class in direct.nn.xpdnet.config) XPDNetEngine (class in direct.nn.xpdnet.xpdnet_engine) Z ZERO_FILLED (direct.nn.types.InitType attribute) zero_pad_to_center() (direct.common.subsample.KtBaseMaskFunc static method) ZEROS (direct.nn.types.InitType attribute) zlimits (direct.data.datasets_config.SheppLoganDatasetConfig attribute)