Please see INSTALL.md.
We provide preprocessed MRSpineSeg and Verse20 datasete. Please use the following links to download them:
- MRSpineSeg (code: 1234)
- Verse20 (code: 1234)
Then unzip them to ./data/dataset .
unzip MRSpineSeg.zip -d ./data/dataset/MRSpineSeg
unzip Verse20.zip -d ./data/dataset/Verse20
- Change the 'model_dir' in ./config.yaml to './data/model/MRSpineSeg';
- Change 'data_path' to './data/MRSpineSeg'
- Run the code:
python train_net.py
- Change the 'model_dir' in ./config.yaml to './data/model/Verse20';
- Change 'data_path' to './data/dataset/Verse20'
- Run the code:
python train_net.py
python test.py