Find a file
2017-08-19 10:59:51 +02:00
.gitignore Created a basic train loop + changed a bit loss and utils 2017-08-17 21:16:19 +02:00
crf.py Added simple eval and test CRF 2017-08-19 10:59:51 +02:00
data_vis.py Created a basic train loop + changed a bit loss and utils 2017-08-17 21:16:19 +02:00
eval.py Added simple eval and test CRF 2017-08-19 10:59:51 +02:00
load.py Created a basic train loop + changed a bit loss and utils 2017-08-17 21:16:19 +02:00
main.py Added simple eval and test CRF 2017-08-19 10:59:51 +02:00
myloss.py Created a basic train loop + changed a bit loss and utils 2017-08-17 21:16:19 +02:00
README.md Initial commit 2017-08-16 14:17:08 +02:00
train.py Added simple eval and test CRF 2017-08-19 10:59:51 +02:00
unet_model.py Created a basic train loop + changed a bit loss and utils 2017-08-17 21:16:19 +02:00
unet_parts.py Added simple eval and test CRF 2017-08-19 10:59:51 +02:00
utils.py Added simple eval and test CRF 2017-08-19 10:59:51 +02:00

Pytorch-UNet

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