The fastest growing cause of blindness is diabetic retinopathy or DR. The purpose of this report was to develop a machine learning model to classify images of eyes according to the degree of diabetic retinopathy. The vision problems caused by this disease are due to high blood sugar causing damage to blood vessels in the retina. Therefore, there are five stages of DR that we attempted to classify including: no disease, mild non-proliferative, moderate non-proliferative, severe non-proliferative, and proliferative. The first model was a convolutional neural network. After building our own network and utilizing transfer learning the best accuracy on the test set was of 0.7166. The second model was a support vector machine which achieved a smaller accuracy of 0.633. The performance of these models does not reach the goal of 98% accuracy but we think it would be difficult to achieve that with this data set. In future projects, we suggest utilizing a data set with more images for each of the stages of the disease. Otherwise, building binary classification models instead of multi-class classification would probably yield better results.
-
Notifications
You must be signed in to change notification settings - Fork 0
The fastest growing cause of blindness is diabetic retinopathy or DB. This occurs when high blood sugar causes damage to blood vessels in the retina causing vision problems. There are four stages of DR, including mild non-proliferative when small areas of the blood vessels swell, moderate non-proliferative when entire blood vessels swell, severe…
mjpramirez/Diabetic-Retinopathy-Prediction
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
The fastest growing cause of blindness is diabetic retinopathy or DB. This occurs when high blood sugar causes damage to blood vessels in the retina causing vision problems. There are four stages of DR, including mild non-proliferative when small areas of the blood vessels swell, moderate non-proliferative when entire blood vessels swell, severe…
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published