Skip to content

GitLab

Commit 2f936b41 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 411653815
parent c13243c1
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
official/vision/beta/projects/panoptic_maskrcnn/tasks/panoptic_maskrcnn.py
View file @ 2f936b41
......@@ -375,12 +375,12 @@ class PanopticMaskRCNNTask(maskrcnn.MaskRCNNTask):
'image_info': labels['image_info']
}
logs.update(
{self.coco_metric.name: (labels['groundtruths'], coco_model_outputs)})
if self._process_iou_metric_on_cpu:
logs.update({
self.coco_metric.name: (labels['groundtruths'], coco_model_outputs),
self.segmentation_perclass_iou_metric.name: (
segmentation_labels,
outputs['segmentation_outputs'])
self.segmentation_perclass_iou_metric.name:
(segmentation_labels, outputs['segmentation_outputs'])
})
else:
self.segmentation_perclass_iou_metric.update_state(
......
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