/probabilistic_unet

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

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Stargazers