Skip to content

This repository contains the code for semantic segmentation of the retina blood vessel on the DRIVE dataset using the PyTorch framework.

Notifications You must be signed in to change notification settings

nikhilroxtomar/Retina-Blood-Vessel-Segmentation-in-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Retina-Blood-Vessel-Segmentation-in-PyTorch

This repository contains the code for semantic segmentation of the retina blood vessel on the DRIVE dataset using the PyTorch framework.

The following models are used:

Models to be used in future:

  • RESUNET
  • DEEPLABV3+
  • more...

Dataset

The ISIC-2018 dataset is used for this for training the UNET architecture. The dataset contains the 2596 pairs of images and masks. All of these images are of different shapes and contains a variety of skin lesions.

Original Image Mask Image

Contact

For any query contact below:

About

This repository contains the code for semantic segmentation of the retina blood vessel on the DRIVE dataset using the PyTorch framework.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages