stefanknegt / Probabilistic-Unet-Pytorch

A Probabilistic U-Net for segmentation of ambiguous images implemented in PyTorch
Apache License 2.0
270 stars 67 forks source link

Predict a test image? #7

Closed voanhkha closed 4 years ago

voanhkha commented 4 years ago

Dear stefanknegt,

I am very happy to see your pytorch implementation of Probabilistic Unet. I am using it for a large-scale research, and later will cite your work in.

May I ask, I have been successfully training the model. But in the validation step, I don't see any script about it. For simplicity, may I ask how can we predict a single image? (is it the output of reconstruct() ? )

Please help me. Thanks so much for your support.

Kind regards, Kha.

jamikapika14 commented 4 years ago

https://github.com/stefanknegt/Probabilistic-Unet-Pytorch/issues/5#issuecomment-502135378