Fix bug for accuracy masking

teamtomo · Jan 3, 2024 · b4c9e43 · b4c9e43
1 parent 5373977
commit b4c9e43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/membrain_seg/segmentation/training/metric_utils.py b/src/membrain_seg/segmentation/training/metric_utils.py
@@ -34,7 +34,7 @@ def masked_accuracy(
     mask = (
         y_gt == ignore_label
         if ignore_label is not None
-        else torch.ones_like(y_gt).bool()
+        else torch.zeros_like(y_gt).bool()
     )
     acc = (threshold_function(y_pred, threshold_value=threshold_value) == y_gt).float()
     acc[mask] = 0.0