File size: 5,201 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 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 |
cd /home/v-qichen3/MRI_recon/code/Frequency-Diffusion/FSMNet/
data_root=/home/v-qichen3/MRI_recon/data/m4raw
python train_m4raw.py --root_path $data_root \
--gpu 3 --batch_size 4 --base_lr 0.0005 --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
--exp FSMNet_m4raw_4x_lr5e-4 --MRIDOWN 4X --MASKTYPE random \
--num_timesteps 30 --image_size 240 --use_kspace --use_time_model
# m4raw dataset, AF=8
# python train_m4raw.py --root_path $data_root \
# --gpu 0 --batch_size 8 --base_lr 0.0001 --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8 \
# --exp FSMNet_m4raw_8x --MRIDOWN 8X --MASKTYPE equispaced \
# --num_timesteps 5 --image_size 240 --use_kspace --use_time_model
# data_root=/bask/projects/j/jiaoj-rep-learn/Hao/datasets/knee
# python train_m4raw.py --root_path $data_root \
# --gpu 1 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.03 --ACCELERATIONS 12 \
# --exp FSMNet_m4raw_12x --MRIDOWN 12X --MASKTYPE equispaced \
# --num_timesteps 30 --image_size 240 --use_kspace --use_time_model
# ---------------- Test ----------------
python test_m4raw.py --root_path $data_root \
--gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
--exp FSMNet_m4raw_4x --phase test --MASKTYPE random --MRIDOWN 4X \
--num_timesteps 30 --image_size 240 --use_kspace --use_time_model --test_tag no_distortion \
--test_sample ColdDiffusion --snapshot_path /home/v-qichen3/MRI_recon/code/Frequency-Diffusion/FSMNet/model/FSMNet_m4raw_4x_lr5e-4_t30_new_kspace_time
#m4raw dataset, AF=8
# python test_m4raw.py --root_path $data_root \
# --gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8 \
# --exp FSMNet_m4raw_8x --phase test --MASKTYPE equispaced --MRIDOWN 8X \
# --num_timesteps 5 --image_size 240 --use_kspace --use_time_model --test_sample Ksample # ColdDiffusion DDPM
# python test_m4raw.py --root_path $data_root \
# --gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.03 --ACCELERATIONS 12 \
# --exp FSMNet_m4raw_12x --phase test --MASKTYPE equispaced --MRIDOWN 12X \
# --num_timesteps 5 --image_size 240 --use_kspace --use_time_model --test_sample Ksample # ColdDiffusion DDPM
# ------------------------------------
# NMSE: 3.2228 ± 0.2863
# PSNR: 28.8534 ± 0.4793
# SSIM: 0.7769 ± 0.0126
# ------------------------------------
# All NMSE: 3.2175 ± 0.5230
# All PSNR: 27.8081 ± 0.7997
# All SSIM: 0.7507 ± 0.0218
# ------------------------------------
# Save Path: model/FSMNet_m4raw_4x_t5_new_kspace_no_distortion//result_case/
# ------------------------------------
# NMSE: 2.9359 ± 0.2239
# PSNR: 29.2540 ± 0.4309
# SSIM: 0.7922 ± 0.0116
# ------------------------------------
# DDPM NMSE: 9.8704 ± 0.5135
# DDPM PSNR: 23.9812 ± 0.2724
# DDPM SSIM: 0.6568 ± 0.0124
# ------------------------------------
# Save Path: model/FSMNet_m4raw_4x_t5_new_kspace_no_distortion_time/result_case/
# ------------------------------------
# NMSE: 2.2551 ± 0.1361
# PSNR: 30.3949 ± 0.4174
# SSIM: 0.8167 ± 0.0126
# ------------------------------------
# Save Path: FSMNet_m4raw_4x_t5_new_kspace_time
# ------------------------------------
# NMSE: 3.8027 ± 0.3095
# PSNR: 27.7977 ± 0.4213
# SSIM: 0.7529 ± 0.0119
# ------------------------------------
# Save Path: FSMNet_m4raw_4x
# ------------------------------------
# NMSE: 6.6898 ± 0.6444
# PSNR: 25.6848 ± 0.4866
# SSIM: 0.6844 ± 0.0149
# ------------------------------------
# ColdDiffusion NMSE: 8.4526 ± 0.7364
# ColdDiffusion PSNR: 24.6651 ± 0.4320
# ColdDiffusion SSIM: 0.6489 ± 0.0141
# ------------------------------------
# Save Path: FSMNet_m4raw_8x_t5_new_kspace_time
# ------------------------------------
# NMSE: 7.8239 ± 0.7702
# PSNR: 24.1138 ± 0.5690
# SSIM: 0.6421 ± 0.0164
# ------------------------------------
# Save Path: FSMNet_m4raw_8x
# ------------------------------------
# NMSE: 7.9649 ± 0.7895
# PSNR: 24.9283 ± 0.4875
# SSIM: 0.6548 ± 0.0150
# ------------------------------------
# All NMSE: 7.9485 ± 1.4481
# All PSNR: 23.8971 ± 0.8682
# All SSIM: 0.6231 ± 0.0348
# ------------------------------------
# Save Path: model/FSMNet_m4raw_8x_t5_new_kspace_no_distortion//result_case/
# ------------------------------------
# NMSE: 7.4375 ± 0.7510
# PSNR: 25.2266 ± 0.4907
# SSIM: 0.6662 ± 0.0140
# ------------------------------------
# All NMSE: 7.4210 ± 1.3740
# All PSNR: 24.1975 ± 0.8648
# All SSIM: 0.6353 ± 0.0317
# ------------------------------------
# Save Path: model/FSMNet_m4raw_8x_t5_new_kspace_time_no_distortion//result_case/
# ------------------------------------
# NMSE: 9.9329 ± 0.8853
# PSNR: 23.9651 ± 0.4525
# SSIM: 0.6380 ± 0.0161
# ------------------------------------
# Save Path: model/FSMNet_m4raw_12x_t5_new_kspace_time//result_case/
# ------------------------------------
# NMSE: 9.4328 ± 0.9025
# PSNR: 23.1375 ± 0.5578
# SSIM: 0.6096 ± 0.0188
# ------------------------------------
# All NMSE: 9.4449 ± 1.9331
# All PSNR: 22.2995 ± 1.0077
# All SSIM: 0.5807 ± 0.0359
# ------------------------------------
# Save Path: model/FSMNet_m4raw_12x//result_case/
|