Skip to content

GitLab

Unverified Commit c15d8a3b authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fixed `image_info` shape bug

parent 037e2b17
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/vision/beta/evaluation/panoptic_quality_evaluator.py
View file @ c15d8a3b
...@@ -164,6 +164,10 @@ class PanopticQualityEvaluator: ...@@ -164,6 +164,10 @@ class PanopticQualityEvaluator:
raise ValueError( raise ValueError(
'Missing the required key `{}` in groundtruths!'.format(k)) 'Missing the required key `{}` in groundtruths!'.format(k))
if isinstance(groundtruths['image_info'], (list, tuple)):
groundtruths['image_info'] = tf.concat(
groundtruths['image_info'], axis=0)
if self._rescale_predictions: if self._rescale_predictions:
for idx in range(len(groundtruths['category_mask'])): for idx in range(len(groundtruths['category_mask'])):
image_info = groundtruths['image_info'][idx] image_info = groundtruths['image_info'][idx]
......
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