Find a file
milesial c8c82204bf First prototype
slow, no gpu, no validation
2017-08-16 14:24:29 +02:00
.gitignore First prototype 2017-08-16 14:24:29 +02:00
data_vis.py First prototype 2017-08-16 14:24:29 +02:00
main.py First prototype 2017-08-16 14:24:29 +02:00
myloss.py First prototype 2017-08-16 14:24:29 +02:00
README.md Initial commit 2017-08-16 14:17:08 +02:00
train.py First prototype 2017-08-16 14:24:29 +02:00
unet_model.py First prototype 2017-08-16 14:24:29 +02:00
unet_parts.py First prototype 2017-08-16 14:24:29 +02:00
utils.py First prototype 2017-08-16 14:24:29 +02:00

Pytorch-UNet

Customized implementation of the U-Net in Pytorch for Kaggle's "Carvana Image Masking Challenge"