项目作者: emadRad
项目描述 :
Segmentation of retinal images for vessel in DRIVE dataset using U-Net
高级语言: Python
项目地址: git://github.com/emadRad/UNet-Baseline.git
UNet-Baseline
Implemntation and fine tuning of the U-Net
model for DRIVE dataset.
Requirements
Python>= 3.5
Pytorch==0.4
Tensorflow==1.12.0 (just for tensorboard)
Running Program
python train.py --config unet_drive.yml