JuliaWolleb / Diffusion-based-Segmentation

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".
MIT License
280 stars 38 forks source link

How can I evaluate the model? #48

Open 648lsf opened 1 year ago

648lsf commented 1 year ago

When I tested, I set an ensemble of 5 runs, but how should the five output pictures I got be merged together to calculate the dice value?

JuliaWolleb commented 1 year ago

Hi

I took the pixel-wise average of the five output pictures, and thresholded it at 0.5.