Retina-Blood-Vessel-Segmentation-in-PyTorch icon indicating copy to clipboard operation
Retina-Blood-Vessel-Segmentation-in-PyTorch copied to clipboard

Published 20 hours ago • verified publisher icon nikhilroxtomar
→

Metadata

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

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:

← Metadata

61
Stars
26
Forks
Watchers

Owner

verified publisher icon nikhilroxtomar

Metadata

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

Back