This is a project for creating a webb app that can classify retina images if it has diabetic retinopathy or not. This was developed as a part of a 24 hour hackathon, Hacktory.
- Python.
- Keras with tensorflow backend.
- Django
With the above requirements,
- Numpy
- Pandas
- Scikit-Learn
- Matplotlib
- Jupyter Notebook
- Clone this repository
- Go to diabetic-retinopathy/website/
- Run 'python3 manage.py runserver'
- Go to localhost/eye in your browser.
- Clone the repository
- Download the dataset, either the subset or the entire dataset.
- Arrange the dataset into Symptoms and Non Symptoms according to Keras ImageDataGenerator.
- Run the Jupyter notebook.
- Model will be saved as model.hd5.
Alternatively, you can use our pretrained model by loading the model.hd5.
The training converged to an accuracy of about 80%. With the validation scores for a small test set generated as,
The training data was generated as,
Since we used the subset of our dataset(scaled down to 64*64) for faster training, we had some limitations in getting higher accuracy. Also using advanced architectures like the inception v3 will boost up our accuracy and help predicting with minimised True negatives.