Skip to content

GitLab

Commit 44b4088c authored by Jiageng Zhang's avatar Jiageng Zhang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 459374753
parent eedfa888
Hide whitespace changes
Inline Side-by-side
Showing with 11 additions and 1 deletion
official/vision/evaluation/panoptic_quality_evaluator.py
View file @ 44b4088c
...@@ -181,4 +181,14 @@ class PanopticQualityEvaluator: ...@@ -181,4 +181,14 @@ class PanopticQualityEvaluator:
self._pq_metric_module.compare_and_accumulate( self._pq_metric_module.compare_and_accumulate(
groundtruths_, predictions_) groundtruths_, predictions_)
else: 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_)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment