projet-long/README.md

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 Soon^TM.