Skip to content

Latest commit

 

History

History
36 lines (29 loc) · 1.57 KB

README.md

File metadata and controls

36 lines (29 loc) · 1.57 KB
Raw

DirectionalFeature

This repository contains the code of the following paper "Learning Directional Feature Maps for Cardiac MRI Segmentation (published in MICCAI2020)", https://arxiv.org/abs/2007.11349

Citation

Please cite the related works in your publications if it helps your research:

@inproceedings{cheng2020learning,
  title={Learning directional feature maps for cardiac mri segmentation},
  author={Cheng, Feng and Chen, Cheng and Wang, Yukang and Shi, Heshui and Cao, Yukun and Tu, Dandan and Zhang, Changzheng and Xu, Yongchao},
  booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
  pages={108--117},
  year={2020},
  organization={Springer}
}

Usage

ACDC Data Preparation

  1. Register and download ACDC-2017 dataset from https://www.creatis.insa-lyon.fr/Challenge/acdc/index.html
  2. Create a folder outside the project with name ACDC_DataSet and copy the dataset.
  3. From the project folder open file acdc_data_preparation.py.
  4. In the file, set the path to ACDC training dataset is pointed as: complete_data_path = '../../ACDC_DataSet/training' .
  5. Run the script acdc_data_preparation.py.
  6. The processed data for training is generated outside the project folder named processed_acdc_dataset.
  7. Run the ./libs/datastes/gen_acdcjson.py to generate the data list for ACDC training and validation.

Training

cd ./tools
python -m torch.distributed.launch --nproc_per_node 4 --master_port $RANDOM train.py --batch_size 12 --mgpus 0,1,2,3 --output_dir logs/... --train_with_eval