项目作者: emadRad

项目描述 :
Segmentation of retinal images for vessel in DRIVE dataset using U-Net
高级语言: Python
项目地址: git://github.com/emadRad/UNet-Baseline.git
创建时间: 2018-11-05T19:06:06Z
项目社区:https://github.com/emadRad/UNet-Baseline

开源协议:

下载


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

  1. python train.py --config unet_drive.yml