Repository for the code used in "Incorporating boundary uncertainty into loss functions for biomedical image segmentation"
The preprint of the paper can be found at: https://arxiv.org/pdf/2111.00533
In this repository, please find the associated Tensorflow/Keras implementation for the following loss functions:
- Dice loss (with boundary uncertainty)
- Dice and cross entropy loss (with boundary uncertainty)
- Unified Focal loss (with boundary uncertainty)