A tutorial reimplementation of Monteiro et al. (2020) on a toy problem.
Open stochastic_segmentation_networks.ipynb and play with the model!
If you are interested in stochastic semantic segmentation you may also want to check out Kassapis et al. (2020) and this toy reimplementation of mine.