File size: 2,360 Bytes
28e6f98 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
cd /home/v-qichen3/MRI_recon/code/Frequency-Diffusion/FSMNet
data_root=/home/v-qichen3/MRI_recon/data/fastmri
python train_fastmri.py --root_path $data_root \
--gpu 2 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
--exp FSMNet_fastmri_4x --MRIDOWN 4X --MASKTYPE random \
--num_timesteps 5 --image_size 320 --use_kspace --use_time_model
# fastMRI dataset, AF=8
# python train_fastmri.py --root_path $data_root \
# --gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8 \
# --exp FSMNet_fastmri_8x --MRIDOWN 8X --MASKTYPE equispaced \
# --num_timesteps 5 --image_size 320 --use_kspace --use_time_model
# python train_fastmri.py --root_path $data_root \
# --gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.03 --ACCELERATIONS 12 \
# --exp FSMNet_fastmri_12x --MRIDOWN 12X --MASKTYPE equispaced \
# --num_timesteps 5 --image_size 320 --use_kspace --use_time_model
# # Test
# #fastMRI dataset, AF=4
# model_4x=model/FSMNet_fastmri_4x/iter_100000.pth
python test_fastmri.py --root_path $data_root \
--gpu 1 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
--exp FSMNet_fastmri_4x --phase test --MRIDOWN 4X \
--num_timesteps 5 --image_size 320 --use_kspace --use_time_model --test_sample Ksample --snapshot_path /home/v-qichen3/MRI_recon/code/Frequency-Diffusion/FSMNet/model/FSMNet_fastmri_4x_t5_kspace_time_no_distortion
# ColdDiffusion DDPM
# #fastMRI dataset, AF=8
python test_fastmri.py --root_path $data_root \
--gpu 3 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8 \
--exp FSMNet_fastmri_8x --phase test --MRIDOWN 8X \
--num_timesteps 5 --image_size 320 --use_kspace --use_time_model --test_sample Ksample --snapshot_path /home/v-qichen3/MRI_recon/code/Frequency-Diffusion/FSMNet/model/FSMNet_fastmri_8x_t5_kspace_time_no_distortion
# ColdDiffusion DDPM
# python test_fastmri.py --root_path $data_root \
# --gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.03 --ACCELERATIONS 12 \
# --exp FSMNet_fastmri_12x --phase test --MRIDOWN 12X \
# --num_timesteps 5 --image_size 320 --use_kspace --use_time_model --test_sample Ksample # ColdDiffusion DDPM
# # FSMNet_fastmri_12x_t30_kspace_time
|