- Notifications
You must be signed in to change notification settings - Fork29
nikhilroxtomar/Retina-Blood-Vessel-Segmentation-in-PyTorch
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
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...
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 |
|---|---|
![]() | ![]() |
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
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Releases
No releases published
Packages0
No packages published

