search

JuliaWolleb / Diffusion-based-Segmentation

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

How do I calculate dice Score and IoU? #60

Open qahsinahk opened 4 months ago

qahsinahk commented 4 months ago

Hello there, If the testing set does not have any segmentation mask (ground truth), then how are we going to calculate the Dice and IoU (Jaccard Index) of the generated samples? There seems to be no code for that at all.

Another issue I got is that the model generates black samples most of the time, I tried to train it a few times, but still had the same problem. Can you please provide me with some solutions. I will be highly grateful. Thank You!