/CANet

TMI 19: CANet: Cross-disease Attention Network for Joint Diabetic Retinopathy and Diabetic Macular Edema Grading

Primary LanguagePython

CANet: Cross-disease Attention Network for Joint Diabetic Retinopathy and Diabetic Macular Edema Grading

Pytorch implementation of CANet: Cross-disease attention network.

Paper

CANet: Cross-disease Attention Network for Joint Diabetic Retinopathy and Diabetic Macular Edema Grading
IEEE Transactions on Medical Imaging, 2019

Installation

  • Install Pytorch 1.1.0 and CUDA 9.0
  • Clone this repo
git clone https://github.com/xmengli999/CANet
cd CANet

Data Preparation

Train

  • Download ImageNet pretrain model and put it under ./pretrain/ or Download the kaggle DR pretrain model and put it under ./pretrain/
  • cd messidor_scripts and specify the pretrain model in train_fold.sh
  • Run sh train_fold.sh to start the training process

Evaluate

  • Specify the model path in eval_fold.sh
  • Run sh eval_fold.sh to start the evaluation.

Citation

If you find the code useful for your research, please cite our paper.

@article{li2019canet,
  title={CANet: Cross-disease Attention Network for Joint Diabetic Retinopathy and Diabetic Macular Edema Grading},
  author={Li, Xiaomeng and Hu, Xiaowei and Yu, Lequan and Zhu, Lei and Fu, Chi-Wing and Heng, Pheng-Ann},
  journal={IEEE transactions on medical imaging},
  year={2019},
  publisher={IEEE}
}

Acknowledgement

CBAM module is reused from the Pytorch implementation of CBAM.

Note