UNet-Baseline

Implemntation and fine tuning of the U-Net model for DRIVE dataset.

Python>= 3.5

Pytorch==0.4

Tensorflow==1.12.0 (just for tensorboard)

python train.py --config unet_drive.yml

Name		Name	Last commit message	Last commit date
Latest commit History 7 Commits
dataloaders		dataloaders
models		models
.gitignore		.gitignore
README.md		README.md
augmentations.py		augmentations.py
logger.py		logger.py
loss.py		loss.py
metrics.py		metrics.py
ops.py		ops.py
optimizers.py		optimizers.py
requirements.txt		requirements.txt
train.py		train.py
unet_drive.yml		unet_drive.yml
utils.py		utils.py

Provide feedback