UNet-Pytorch
Simple And Friendly
Installation
- Install Pytorch from http://pytorch.org/
- Install Torchvision from the source
git clone https://github.com/pytorch/vision
cd vision
python setup.py install
Data preparation
The Carvana data is available on the Kaggle website.
The input images and target masks should be in the data/train
and data/test
folders respectively.
Train & Test & Evaluation & Visualization
Just following UNet-Pytorch.ipynb~