All workPRJ-08
Machine Learning

ROP Classifier

ML / Computer Vision · 2025

Machine LearningML / Computer Vision
Overview

A computer vision model trained to classify Retinopathy of Prematurity from clinical retinal images.

Role

ML / Computer Vision

Timeline

4 months

Year

2025

Stack

PyTorch · OpenCV

The Challenge

Clinical retinal images vary a lot in quality, and a classifier trained on raw images does not hold up. The baseline needed to be beaten on the images that actually matter.

Approach

I built image preprocessing and segmentation pipelines ahead of the classifier so the model trained on cleaned, region-focused data, then measured against the baseline.

  • Image preprocessing pipeline for variable-quality retinal images
  • Segmentation pipeline isolating the relevant regions
  • Classification accuracy improved over the baseline
Outcome

Accuracy improved over the baseline through the preprocessing and segmentation stages.

Built with
PyTorchOpenCV
Next projectExotic Mahseer