├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── algorithms ├── __init__.py └── dps.py ├── conda_env.yml ├── configs ├── different_steps │ ├── Brain_2D_R10_1000.yml │ ├── Brain_2D_R10_1000_Schurn100.yml │ ├── Brain_2D_R10_1000_Schurn150.yml │ ├── Brain_2D_R10_1000_Schurn200.yml │ ├── Brain_2D_R10_1000_Schurn250.yml │ ├── Brain_2D_R10_1000_Schurn300.yml │ ├── Brain_2D_R10_1000_Schurn350.yml │ ├── Brain_2D_R10_1000_Schurn400.yml │ ├── Brain_2D_R10_1000_Schurn50.yml │ ├── Brain_2D_R10_200.yml │ ├── Brain_2D_R10_25.yml │ ├── Brain_2D_R10_400.yml │ ├── Brain_2D_R10_600.yml │ └── Brain_2D_R10_800.yml ├── new_brains │ ├── Brain_2D_R10_L2-1-200train.yml │ ├── Brain_2D_R12_L2-1-200train.yml │ ├── Brain_2D_R2_L2-1-200train.yml │ ├── Brain_2D_R4_L2-1-200train.yml │ ├── Brain_2D_R6_L2-1-200train.yml │ ├── Brain_2D_R8_L2-1-200train.yml │ ├── Brain_3D_R10_L2-1-200train.yml │ ├── Brain_3D_R12_L2-1-200train.yml │ ├── Brain_3D_R14_L2-1-200train.yml │ ├── Brain_3D_R16_L2-1-200train.yml │ ├── Brain_3D_R18_L2-1-200train.yml │ ├── Brain_3D_R20_L2-1-200train.yml │ ├── Brain_3D_R2_L2-1-200train.yml │ ├── Brain_3D_R4_L2-1-200train.yml │ ├── Brain_3D_R6_L2-1-200train.yml │ └── Brain_3D_R8_L2-1-200train.yml └── new_knees │ ├── Knee_2D_R10_L2-1-200train.yml │ ├── Knee_2D_R12_L2-1-200train.yml │ ├── Knee_2D_R2_L2-1-200train.yml │ ├── Knee_2D_R4_L2-1-200train.yml │ ├── Knee_2D_R6_L2-1-200train.yml │ ├── Knee_2D_R8_L2-1-200train.yml │ ├── Knee_3D_R10_L2-1-200train.yml │ ├── Knee_3D_R12_L2-1-200train.yml │ ├── Knee_3D_R14_L2-1-200train.yml │ ├── Knee_3D_R16_L2-1-200train.yml │ ├── Knee_3D_R18_L2-1-200train.yml │ ├── Knee_3D_R20_L2-1-200train.yml │ ├── Knee_3D_R2_L2-1-200train.yml │ ├── Knee_3D_R4_L2-1-200train.yml │ ├── Knee_3D_R6_L2-1-200train.yml │ └── Knee_3D_R8_L2-1-200train.yml ├── datasets ├── __init__.py └── mri_dataloaders.py ├── learners ├── __init__.py ├── baseline_mask.py ├── gradientlearner.py └── probabilistic_mask.py ├── main.py ├── problems ├── __init__.py └── fourier_multicoil.py └── utils ├── __init__.py ├── exp_utils.py └── metric_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/README.md -------------------------------------------------------------------------------- /algorithms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/dps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/algorithms/dps.py -------------------------------------------------------------------------------- /conda_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/conda_env.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_1000.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_1000.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_1000_Schurn100.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_1000_Schurn100.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_1000_Schurn150.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_1000_Schurn150.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_1000_Schurn200.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_1000_Schurn200.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_1000_Schurn250.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_1000_Schurn250.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_1000_Schurn300.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_1000_Schurn300.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_1000_Schurn350.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_1000_Schurn350.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_1000_Schurn400.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_1000_Schurn400.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_1000_Schurn50.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_1000_Schurn50.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_200.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_200.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_25.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_25.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_400.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_400.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_600.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_600.yml -------------------------------------------------------------------------------- /configs/different_steps/Brain_2D_R10_800.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/different_steps/Brain_2D_R10_800.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_2D_R10_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_2D_R10_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_2D_R12_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_2D_R12_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_2D_R2_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_2D_R2_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_2D_R4_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_2D_R4_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_2D_R6_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_2D_R6_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_2D_R8_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_2D_R8_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_3D_R10_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_3D_R10_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_3D_R12_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_3D_R12_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_3D_R14_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_3D_R14_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_3D_R16_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_3D_R16_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_3D_R18_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_3D_R18_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_3D_R20_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_3D_R20_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_3D_R2_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_3D_R2_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_3D_R4_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_3D_R4_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_3D_R6_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_3D_R6_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_brains/Brain_3D_R8_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_brains/Brain_3D_R8_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_2D_R10_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_2D_R10_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_2D_R12_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_2D_R12_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_2D_R2_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_2D_R2_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_2D_R4_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_2D_R4_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_2D_R6_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_2D_R6_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_2D_R8_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_2D_R8_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_3D_R10_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_3D_R10_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_3D_R12_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_3D_R12_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_3D_R14_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_3D_R14_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_3D_R16_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_3D_R16_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_3D_R18_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_3D_R18_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_3D_R20_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_3D_R20_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_3D_R2_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_3D_R2_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_3D_R4_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_3D_R4_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_3D_R6_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_3D_R6_L2-1-200train.yml -------------------------------------------------------------------------------- /configs/new_knees/Knee_3D_R8_L2-1-200train.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/configs/new_knees/Knee_3D_R8_L2-1-200train.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/mri_dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/datasets/mri_dataloaders.py -------------------------------------------------------------------------------- /learners/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /learners/baseline_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/learners/baseline_mask.py -------------------------------------------------------------------------------- /learners/gradientlearner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/learners/gradientlearner.py -------------------------------------------------------------------------------- /learners/probabilistic_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/learners/probabilistic_mask.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/main.py -------------------------------------------------------------------------------- /problems/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /problems/fourier_multicoil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/problems/fourier_multicoil.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/exp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/utils/exp_utils.py -------------------------------------------------------------------------------- /utils/metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sriram-Ravula/MRI_Sampling_Diffusion/HEAD/utils/metric_utils.py --------------------------------------------------------------------------------