dev(MeanIoU): add NOTE in self.add

open-mmlab · Sep 13, 2022 · 3a03812 · 3a03812
1 parent 06d9348
commit 3a03812
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmeval/segmentation/mean_iou.py b/mmeval/segmentation/mean_iou.py
@@ -122,6 +122,8 @@ def add(self, predictions: Sequence, labels: Sequence) -> None: # type: ignore
  predictions (Sequence): A batch of outputs from the model.
  """
  for prediction, label in zip(predictions, labels):
+ # NOTE: We need to store the confusion matrix from each sample so
+ # that we can remove pad samples after all-gather.
  cm = self.compute_confusion_matrix(prediction, label,
  self.num_classes)
  self._results.append(cm)