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/