qic999's picture
Upload folder using huggingface_hub
28e6f98 verified
Raw
History Blame Contribute Delete
1.08 kB
# fastMRI dataset, AF=4
# BraTS dataset, AF=4
mamba activate diffmri
cd /home/cbtil3/hao/repo/Frequency-Diffusion/experiments/FSMNet
data_root=/bask/projects/j/jiaoj-rep-learn/Hao/datasets/knee
#data_root=/gamedrive/Datasets/medical/FrequencyDiffusion
python train_fastmri.py --root_path $data_root \
--gpu 0 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
--exp FSMNet_fastmri_4x
# fastMRI dataset, AF=8
python train_fastmri.py --root_path /data/qic99/MRI_recon/fastMRI/ \
--gpu 1 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8 \
--exp FSMNet_fastmri_8x
# Test
fastMRI dataset, AF=4
python test_fastmri.py --root_path /data/qic99/MRI_recon/fastMRI/ \
--gpu 5 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.08 --ACCELERATIONS 4 \
--exp FSMNet_fastmri_4x --phase test
fastMRI dataset, AF=8
python test_fastmri.py --root_path /data/qic99/MRI_recon/fastMRI/ \
--gpu 6 --batch_size 4 --base_lr 0.0001 --CENTER_FRACTIONS 0.04 --ACCELERATIONS 8 \
--exp FSMNet_fastmri_8x --phase test