Add surface loss in keras

LIVIAETS / boundary-loss

Official code for "Boundary loss for highly unbalanced segmentation", runner-up for best paper award at MIDL 2019. Extended version in MedIA, volume 67, January 2021.

https://doi.org/10.1016/j.media.2020.101851

MIT License

654 stars 98 forks source link

Add surface loss in keras #19

Closed akamojo closed 4 years ago

akamojo commented 5 years ago

Together with @marcinkaczor, we added a simple script which includes functions for calculating surface loss using keras. Feel free to modify it according to your needs.

HKervadec commented 5 years ago

Great ! Thanks for the contribution.

I will review soon (will be mostly in a plane for the next few days), I will think about the best way to integrate it in the current repository. Modifying from binary to multi-class should be trivial also, but perhaps I will end up with two separate functions.