Supervised Uncertainty Quantification for Segmentation with Multiple Annotations

The PyTorch implementation of the paper: Supervised Uncertainty Quantification for Segmentation with Multiple Annotations.

Requirements

This code is tested on:

• Python 3.6
• PyTorch 1.1.0
• Nvidia TITAN Xp GPU

Acknowledgments

The PyTorch implementation of the Probabilistic U-Net is modified from: https://github.com/stefanknegt/Probabilistic-Unet-Pytorch.