Find a file
milesial 3c14fc3e63 Final tweaks
Former-commit-id: 547d4580c776afa5782a08fc5288526da82a0972
2017-09-26 21:00: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 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 Final tweaks 2017-09-26 21:00:51 +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 Final tweaks 2017-09-26 21:00:51 +02:00

Pytorch-UNet

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