Skip to content

GitLab

Commit 496a77e8 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 445515247
parent 89ee0f7c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
official/vision/beta/projects/panoptic_maskrcnn/serving/panoptic_segmentation.py
View file @ 496a77e8
...@@ -129,9 +129,9 @@ class PanopticSegmentationModule(detection.DetectionModule): ...@@ -129,9 +129,9 @@ class PanopticSegmentationModule(detection.DetectionModule):
scores = tf.nn.softmax(masks, axis=-1) scores = tf.nn.softmax(masks, axis=-1)
final_outputs.update({ final_outputs.update({
'detection_masks': detections['detection_masks'], 'detection_masks': detections['detection_masks'],
'masks': masks, 'semantic_logits': masks,
'scores': scores, 'semantic_scores': scores,
'classes': classes, 'semantic_classes': classes,
'image_info': image_info 'image_info': image_info
}) })
if model_params.generate_panoptic_masks: if model_params.generate_panoptic_masks:
......
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