Skip to content

Latest commit

 

History

History
17 lines (11 loc) · 336 Bytes

README.md

File metadata and controls

17 lines (11 loc) · 336 Bytes

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