Update dice.py

constantinpape · Oct 17, 2023 · 40c2e09 · 40c2e09
1 parent 6ba60db
commit 40c2e09
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/elf/evaluation/dice.py b/elf/evaluation/dice.py
@@ -8,6 +8,9 @@
 def dice_score(segmentation, groundtruth, threshold_seg=0, threshold_gt=0):
     """ Compute the dice score between binarized segmentation and ground-truth.
 
+    Note: for comparing probaility maps (i.e. predictions in range [0, 1]) with this function
+    you need to set the thresholds to None. Otherwise the results will be wrong.
+
     Arguments:
         segmentation [np.ndarray] - candidate segmentation to evaluate
         groundtruth [np.ndarray] - groundtruth