/2DUNet

2DUNet

Primary LanguagePython

2D UNet

2DUNet implementation https://arxiv.org/abs/1505.04597

Data needs to be organised in format

data/ train_images/ train_masks/ val_images/ val_masks/

Create a folder saved_images for the predictions to be saved

The model was trained and tested using Carvana dataset from Kaggle