probabilistic_unet. A U-Net combined with a variational auto-encoder that is able to learn conditional distributions over semantic segmentations.