rpytel1 / efficient-semantic-segmentation

2 stars 1 forks source link

Experiment with losses #7

Closed rpytel1 closed 4 years ago

rpytel1 commented 4 years ago
  1. add comfortable implementation to add new losses
  2. try implemented losses and compare results.
rpytel1 commented 4 years ago

Implemented: Lovasz and Dice loss. Best performance still for Cross-correlation but for IOU it shows that there is clearly some unbalanced data. Lovasz: Better performance after 12 epochs. Probably could get better for more epochs. Dice: Better IOU performance but not so good accuracy.

Tests performed on Dynamic UNet with Resnet backbone.