Probabilistic-Unet-Pytorch icon indicating copy to clipboard operation
Probabilistic-Unet-Pytorch copied to clipboard
verified publisher icon stefanknegt

Sampling a Segmentation

Open anwai98 opened this issue 3 years ago • 0 comments

Hi Stefan, thanks for this nice PyTorch implementation. I have a question regarding the sample function in the probabilistic_unet.py file. In line 222 (while training), wouldn't we want to sample from the posterior net? (according to the paper's training heuristic, correct me if I am wrong)

anwai98 avatar Nov 03 '22 09:11 anwai98