Diabetic Retinopathy Detection using convolutional neural networks
Dataset available in Kaggle
Data labels
- 0 - No DR
- 1 - Mild
- 2 - Moderate
- 3 - Severe
- 4 - Proliferative DR
Setup
- Install virtualenv and python3
- virtualenv -p python3 env
- source env/bin/activate
- pip install -r requirements.txt
How to run
- python train.py
- python predict.py