Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-11-09 15:02:03 +00:00
Code Issues Packages Projects Releases Wiki Activity
REVA-QCAV/README.md
milesial e1bf150da3 Updated README.md 
Former-commit-id: 03065569518394e7b686e6995d05599f9822417f
2017-11-30 03:44:29 +01:00

857 B
Raw Blame History

Pytorch-UNet

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

This model scored a dice coefficient of 0.988423 (511 out of 735), which is bad but could be improved with more training, data augmentation, fine tuning, and playing with CRF post-processing.

The model used for the last submission is stored in the MODEL.pth file, if you wish to play with it. The data is available on the Kaggle website.

Note

The code and the overall project architecture is a big mess for now, as I left it abandoned when the challenge finished. I will clean it SoonTM.