zmbhou/CANet

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

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

• Download Messidor dataset
  
• Put the data under ./data/

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

• Contact: Xiaomeng Li (xmengli999@gmail.com)