Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "1ec7ff73d48af07403a74613ed5756b14a2630b6"
Commit f24c95ae authored by Xianzhi Du's avatar Xianzhi Du Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 382408293
parent edc65712
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
official/vision/beta/serving/detection.py
View file @ f24c95ae
...@@ -162,9 +162,7 @@ class DetectionModule(export_base.ExportModule): ...@@ -162,9 +162,7 @@ class DetectionModule(export_base.ExportModule):
else: else:
final_outputs = { final_outputs = {
'decoded_boxes': detections['decoded_boxes'], 'decoded_boxes': detections['decoded_boxes'],
'decoded_box_scores': detections['decoded_box_scores'], 'decoded_box_scores': detections['decoded_box_scores']
'cls_outputs': detections['cls_outputs'],
'box_outputs': detections['box_outputs']
} }
if 'detection_masks' in detections.keys(): if 'detection_masks' in detections.keys():
......
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