Goto

Collaborating Authors

 milesial pytorch-unet


milesial/Pytorch-UNet

#artificialintelligence

This model was trained from scratch with 5000 images (no data augmentation) and scored a dice coefficient of 0.988423 (511 out of 735) on over 100k test images. This score could be improved with more training, data augmentation, fine tuning, playing with CRF post-processing, and applying more weights on the edges of the masks. The Carvana data is available on the Kaggle website. You can easily test the output masks on your images via the CLI. You can specify which model file to use with --model MODEL.pth.