Pytorch Bayesian UNet model for segmentation and uncertainty prediction - View it on GitHub