Internal change

PiperOrigin-RevId: 346129282

Internal change
PiperOrigin-RevId: 346129282
84dc1fd6 · A. Unique TensorFlower · fd270433 · 84dc1fd6
Commit 84dc1fd6 authored Dec 07, 2020 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/beta/serving/detection.py official/vision/beta/serving/detection.py +1 -1

No files found.
--- a/official/vision/beta/serving/detection.py
+++ b/official/vision/beta/serving/detection.py
@@ -135,6 +135,6 @@ class DetectionModule(export_base.ExportModule):
        'num_detections': detections['num_detections']
    }
    if 'detection_masks' in detections.keys():
-      final_outputs.update('detection_masks', detections['detection_masks'])
+      final_outputs['detection_masks'] = detections['detection_masks']

    return final_outputs