awsmak/retina_blood_vessel_segmentation

Retina Blood Vessel Segmentation using UNET on PyTorch

This is an implementation of UNET on Python 3, for semantic segmentation of the retina blood vessel on the DRIVE dataset using the PyTorch framework.

• Model : UNET
• Dataset : https://www.kaggle.com/datasets/zionfuo/drive2004

Dataset

image | mask