Fix fence(IoU) = 0 when training on PascalContextDataset59;

open-mmlab · Apr 11, 2021 · 2cffeaf · 2cffeaf
1 parent e0e985f
commit 2cffeaf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mmseg/core/evaluation/metrics.py b/mmseg/core/evaluation/metrics.py
@@ -57,11 +57,11 @@ def intersect_and_union(pred_label,
 
     intersect = pred_label[pred_label == label]
     area_intersect = torch.histc(
-        intersect.float(), bins=(num_classes), min=0, max=num_classes)
+        intersect.float(), bins=(num_classes), min=0, max=num_classes - 1)
     area_pred_label = torch.histc(
-        pred_label.float(), bins=(num_classes), min=0, max=num_classes)
+        pred_label.float(), bins=(num_classes), min=0, max=num_classes - 1)
     area_label = torch.histc(
-        label.float(), bins=(num_classes), min=0, max=num_classes)
+        label.float(), bins=(num_classes), min=0, max=num_classes - 1)
     area_union = area_pred_label + area_label - area_intersect
     return area_intersect, area_union, area_pred_label, area_label