Internal change

PiperOrigin-RevId: 515246811

Author: fyangf

official/vision/serving/detection.py

@@ -203,5 +203,6 @@ def serve(self, images: tf.Tensor):
     if 'detection_masks' in detections.keys():
       final_outputs['detection_masks'] = detections['detection_masks']
 
-    final_outputs.update({'image_info': image_info})
+    if self.params.task.model.detection_generator.nms_version != 'tflite':
+      final_outputs.update({'image_info': image_info})
     return final_outputs