From e1bf150da3b8641396c24f77d264923122092286 Mon Sep 17 00:00:00 2001 From: milesial Date: Thu, 30 Nov 2017 03:44:29 +0100 Subject: [PATCH] Updated README.md Former-commit-id: 03065569518394e7b686e6995d05599f9822417f --- README.md | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 8e033e5..f70eb88 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,11 @@ # Pytorch-UNet -Customized implementation of the U-Net in Pytorch for Kaggle's "Carvana Image Masking Challenge" +Customized implementation of the [U-Net](https://arxiv.org/pdf/1505.04597.pdf) in Pytorch for Kaggle's [Carvana Image Masking Challenge](https://www.kaggle.com/c/carvana-image-masking-challenge). + +This model scored a [dice coefficient](https://en.wikipedia.org/wiki/S%C3%B8rensen%E2%80%93Dice_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](https://www.kaggle.com/c/carvana-image-masking-challenge/data). + + + +## 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.