Skip to content

Subhranil2004/retinal-fundus-segmentation

Repository files navigation

Retinal Fundus Image Segmentation

This repository contains code for segmenting retinal fundus images using deep learning techniques. The segmentation process involves identifying and isolating specific structures within the retinal images, such as blood vessels, optic discs, and lesions.

Currently, the main focus of segmentation is on identifying the lesions. The segmentation is performed using an U-Net model, trained on a dataset of retinal fundus images and their corresponding masks.

The segmentation process is implemented in a Jupyter notebook (segmentation.ipynb), which includes data preprocessing, model training, and visualization of results.
The loss function used for training the model is a combination of Binary Cross-Entropy (BCE) and Dice Loss, which helps in improving the accuracy of segmentation.

Dataset samples

Retinal Fundus Images Corresponding Annotations
Sample Image 1 Annotation 1
Sample Image 2 Annotation 2

Results so far

  • Recall: 82.87%
  • Precision: 76.14%
  • F1-Score: 79.34%

Visualization of Results

Segmentation Results Segmentation Results Segmentation Results

Future Work

  • Remove optic disc from the images to reduce false positives.
  • Experiment with different architectures and hyperparameters to improve segmentation accuracy.
  • Post-processing techniques to refine the segmented masks.

Note

A webapp for this segmentation model is under development and will be added soon.


Suggestions and contributions are welcome! 🙂

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published