Early-Stage Diabetic Retinopathy Prediction Using Advanced Machine Learning

Project Overview Diabetic Retinopathy (DR) is one of the leading causes of preventable blindness among diabetic patients worldwide. Early detection of Diabetic Retinopathy is essential to prevent vision loss and reduce treatment costs.

This project develops an AI-powered system that analyzes retinal fundus images to detect early-stage Diabetic Retinopathy using advanced deep learning techniques. The system applies multiple convolutional neural network (CNN) models and handles class imbalance commonly present in medical datasets.

The solution includes a machine learning pipeline, a backend API for inference, and a frontend interface to upload retinal images and obtain predictions.

Objectives Detect early-stage Diabetic Retinopathy using retinal images. Apply advanced deep learning models for medical image classification. Handle class imbalance in medical datasets. Provide an AI-powered decision support system for DR screening. Develop a deployable system using FastAPI backend and web frontend.

Dataset This project uses the APTOS 2019 Blindness Detection Dataset. Dataset Source: https://www.kaggle.com/competitions/aptos2019-blindness-detection/data The dataset contains retinal fundus images labeled according to the severity of diabetic retinopathy.

Class Labels Label Description 0 No Diabetic Retinopathy 1 Mild DR (Early Stage) 2 Moderate DR 3 Severe DR 4 Proliferative DR Images are high-resolution retinal scans used for medical diagnosis.

Machine Learning Pipeline

Image Preprocessing

Resize images to 224×224 Data augmentation Normalization Deep Learning Models

The system uses a multi-model architecture: ResNet50 Efficientnet-B0

Built With

  • efficientnet-b0
  • fastapi
  • html/css
  • kaggle
  • opencv
  • python
  • resnet50
  • tensorflow/keras
Share this project:

Updates