Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "1af6bf47681304b272650bb59310b3cfa41b347b"
Commit 84dc1fd6 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 346129282
parent fd270433
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/serving/detection.py
View file @ 84dc1fd6
...@@ -135,6 +135,6 @@ class DetectionModule(export_base.ExportModule): ...@@ -135,6 +135,6 @@ class DetectionModule(export_base.ExportModule):
'num_detections': detections['num_detections'] 'num_detections': detections['num_detections']
} }
if 'detection_masks' in detections.keys(): if 'detection_masks' in detections.keys():
final_outputs.update('detection_masks', detections['detection_masks']) final_outputs['detection_masks'] = detections['detection_masks']
return final_outputs return final_outputs
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