Based on milesial/Pytorch-UNet/ Used to train models for paint detection and mask generation on training dataset of 192 image/mask combos.
Sample prediction:
- Image name: f17x2022_06_28.mp4.track000206.frame006589.jpg
- Dice coefficient: 0.9417218543046357
Green = True Mask, Blue = Predicted Mask, Red = Overlap
Training visualized using the weights and balances library and web api:
