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