Boundary uncertainty

Repository for the code used in "Incorporating boundary uncertainty into loss functions for biomedical image segmentation"

Source

The preprint of the paper can be found at: https://arxiv.org/pdf/2111.00533

Description of repository contents

In this repository, please find the associated Tensorflow/Keras implementation for the following loss functions:

1. Dice loss (with boundary uncertainty)
2. Dice and cross entropy loss (with boundary uncertainty)
3. Unified Focal loss (with boundary uncertainty)