qic999's picture
Upload folder using huggingface_hub
28e6f98 verified
Raw
History Blame Contribute Delete
1.21 kB
# BraTS dataset, AF=4
mamba activate diffmri
cd /home/cbtil3/hao/repo/Frequency-Diffusion/experiments/FSMNet
#root_path=/bask/projects/j/jiaoj-rep-learn/Hao/datasets/knee/image_100patients_4X/
root_path=/gamedrive/Datasets/medical/FrequencyDiffusion/brats/image_100patients_4X/
python train_brats.py --root_path $root_path\
--gpu 0 --batch_size 4 --base_lr 0.0001 --MRIDOWN 4X --low_field_SNR 0 \
--input_normalize mean_std \
--exp FSMNet_BraTS_4x
BraTS dataset, AF=8
python train_brats.py --root_path /gamedrive/Datasets/medical/FrequencyDiffusion/brats/image_100patients_8X/ \
--gpu 1 --batch_size 4 --base_lr 0.0001 --MRIDOWN 8X --low_field_SNR 0 \
--input_normalize mean_std \
--exp FSMNet_BraTS_8x
# Test
BraTS dataset, AF=4
python test_brats.py --root_path /data/qic99/MRI_recon/image_100patients_4X/ \
--gpu 3 --base_lr 0.0001 --MRIDOWN 4X --low_field_SNR 0 \
--input_normalize mean_std \
--exp FSMNet_BraTS_4x --phase test
BraTS dataset, AF=8
python test_brats.py --root_path /data/qic99/MRI_recon/image_100patients_8X/ \
--gpu 4 --base_lr 0.0001 --MRIDOWN 8X --low_field_SNR 0 \
--input_normalize mean_std \
--exp FSMNet_BraTS_8x --phase test