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/