Find a file
milesial 932e1c3cf5 Added simple trained model
Former-commit-id: e87526c043c447e5c8d95ee8314fcb7f97c13ce2
2017-08-23 17:38:55 +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 predict + submit script 2017-08-21 18:00:07 +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 predict + submit script 2017-08-21 18:00:07 +02:00
load.py Added simple predict + submit script 2017-08-21 18:00:07 +02:00
main.py Added simple eval and test CRF 2017-08-19 10:59:51 +02:00
MODEL.pth.REMOVED.git-id Added simple trained model 2017-08-23 17:38:55 +02:00
myloss.py Created a basic train loop + changed a bit loss and utils 2017-08-17 21:16:19 +02:00
predict.py Added simple predict + submit script 2017-08-21 18:00:07 +02:00
README.md Initial commit 2017-08-16 14:17:08 +02:00
submit.py Added simple predict + submit script 2017-08-21 18:00:07 +02:00
train.py Added simple trained model 2017-08-23 17:38:55 +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 predict + submit script 2017-08-21 18:00:07 +02:00

Pytorch-UNet

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