Internal change

PiperOrigin-RevId: 346129282

Internal change
PiperOrigin-RevId: 346129282
7f9bf1f2 · A. Unique TensorFlower · 69845d55 · 7f9bf1f2
Commit 7f9bf1f2 authored Dec 07, 2020 by A. Unique TensorFlower
Show 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