Merge remote-tracking branch 'upstream/master'

578b320d · Gunho Park · 504f2550 · 44b4088c · 578b320d
Commit 578b320d authored Jul 07, 2022 by Gunho Park
Show whitespace changes
Inline Side-by-side

Showing with 11 additions and 1 deletion

official/vision/evaluation/panoptic_quality_evaluator.py official/vision/evaluation/panoptic_quality_evaluator.py +11 -1

No files found.
--- a/official/vision/evaluation/panoptic_quality_evaluator.py
+++ b/official/vision/evaluation/panoptic_quality_evaluator.py
@@ -181,4 +181,14 @@ class PanopticQualityEvaluator:
        self._pq_metric_module.compare_and_accumulate(
            groundtruths_, predictions_)
    else:
-      self._pq_metric_module.compare_and_accumulate(groundtruths, predictions)
+      for idx in range(len(groundtruths['category_mask'])):
+        groundtruths_ = {
+            'category_mask': groundtruths['category_mask'][idx],
+            'instance_mask': groundtruths['instance_mask'][idx]
+        }
+        predictions_ = {
+            'category_mask': predictions['category_mask'][idx],
+            'instance_mask': predictions['instance_mask'][idx]
+        }
+        self._pq_metric_module.compare_and_accumulate(groundtruths_,
+                                                      predictions_)