adapt dice score for empty gt mask

wasserth · Feb 6, 2024 · 350bb89 · 350bb89
1 parent deafa2d
commit 350bb89
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_end_to_end.py b/tests/test_end_to_end.py
@@ -10,6 +10,8 @@
 
 
 def dice_score(y_true, y_pred):
+    if y_true.sum() == 0 and y_pred.sum() == 0:
+        return 1.0
     intersect = np.sum(y_true * y_pred)
     denominator = np.sum(y_true) + np.sum(y_pred)
     f1 = (2 * intersect) / (denominator + 1e-6)