# BraTS dataset, AF=4
mamba activate diffmri
cd /home/cbtil3/hao/repo/Frequency-Diffusion/FSMNet-modify
cd /bask/projects/j/jiaoj-rep-learn/Hao/repo/Frequency-Diffusion/FSMNet-modify


gamedrive=/media/cbtil3/74ec35fd-2452-4dcc-8d7d-3ba957e302c9

#4T folder: /media/cbtil3/9feaf350-913e-4def-8114-f03573c04364/hao
root_path_4x=/bask/projects/j/jiaoj-rep-learn/Hao/datasets/knee/image_100patients_4X/
root_path_8x=/bask/projects/j/jiaoj-rep-learn/Hao/datasets/knee/image_100patients_8X/

root_path_4x=$gamedrive/Datasets/medical/FrequencyDiffusion/brats/image_100patients_4X/
root_path_8x=$gamedrive/Datasets/medical/FrequencyDiffusion/brats/image_100patients_8X/

python train_brats.py --root_path $root_path_4x\
    --gpu 0 --batch_size 4 --base_lr 0.0001 --MRIDOWN 4X --low_field_SNR 0 \
    --input_normalize mean_std --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
    --exp FSMNet_BraTS_4x  --use_time_model --use_kspace

# BraTS dataset, AF=8
python train_brats.py --root_path $root_path_8x \
    --gpu 0 --batch_size 4 --base_lr 0.0001 --MRIDOWN 8X --low_field_SNR 0 \
    --input_normalize mean_std --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8\
    --exp FSMNet_BraTS_8x  --use_time_model --use_kspace




# Test
#BraTS dataset, AF=4

python test_brats.py --root_path $root_path_4x \
    --gpu 0 --base_lr 0.0001 --MRIDOWN 4X --low_field_SNR 0 \
    --input_normalize mean_std --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4  \
    --exp FSMNet_BraTS_4x --phase test   --use_time_model --use_kspace



#BraTS dataset, AF=8

python test_brats.py --root_path $root_path_8x  \
    --gpu 1 --base_lr 0.0001 --MRIDOWN 8X --low_field_SNR 0 \
    --input_normalize mean_std --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8  \
    --exp FSMNet_BraTS_8x --phase test  --use_time_model --use_kspace