The PyTorch implementation of the paper: Supervised Uncertainty Quantification for Segmentation with Multiple Annotations.
This code is tested on:
- Python 3.6
- PyTorch 1.1.0
- Nvidia TITAN Xp GPU
The PyTorch implementation of the Probabilistic U-Net is modified from: https://github.com/stefanknegt/Probabilistic-Unet-Pytorch.