This repository contains the Keras implementation using Tensorflow as backend of our paper "CABNet: Category Attention Block for Imbalanced Diabetic Retinopathy Grading"
python 3.6
numpy 1.16.4
keras 2.3.1
tensorflow 1.13.1
pillow 7.0.0
opencv-python 4.1.0
The trained model : Baidu Yun : https://pan.baidu.com/s/1G2tKYNvqP7jmCl-svC1cfA password: kjga
Google Drive:https://drive.google.com/drive/folders/1GNn3tj7WTLPjdOJUZqItui58DDQtYrsS?usp=sharing
Clone the repository, and download the trained model, put them into 'weights' folder, you can run test.py to test the model directly. The details of the trained model are listed in details of trained models.txt file.
If you want to train the model, download the dataset and put them intodata
folder.
And then run the code:python train.py Note that the parameters and paths should be set beforehand
Once the training is complete, you can run the test.py to test your model. Run the code : python test.py.
Code can only be used for ACADEMIC PURPOSES. NO COMERCIAL USE is allowed. Copyright © College of Computer Science, Nankai University. All rights reserved.
@article{he2020cabnet,
title={CABNet: Category attention block for imbalanced diabetic retinopathy grading},
author={He, Along and Li, Tao and Li, Ning and Wang, Kai and Fu, Huazhu},
journal={IEEE Transactions on Medical Imaging},
volume={40},
number={1},
pages={143--153},
year={2020},
publisher={IEEE}
}