Internal change

PiperOrigin-RevId: 506457533

Author: fyangf

official/vision/configs/common.py

@@ -145,3 +145,6 @@ class TFLitePostProcessingConfig(hyperparams.Config):
   use_regular_nms: bool = False
   nms_score_threshold: float = 0.1
   nms_iou_threshold: float = 0.5
+  # Whether to normalize coordinates of anchors to [0, 1]. If setting to True,
+  # coordinates of output boxes is also normalized but latency increases.
+  normalize_anchor_coordinates: Optional[bool] = False

official/vision/modeling/layers/detection_generator.py

@@ -533,11 +533,13 @@ def _generate_detections_tflite(raw_boxes: Mapping[str, tf.Tensor],
   wa = anchors[..., 3] - anchors[..., 1]
   anchors = tf.stack([ycenter_a, xcenter_a, ha, wa], axis=-1)
 
-  # TFLite's object detection APIs require normalized anchors.
-  height, width = config['input_image_size']
-  normalize_factor = tf.constant([height, width, height, width],
-                                 dtype=tf.float32)
-  anchors = anchors / normalize_factor
+  if config.get('normalize_anchor_coordinates', False):
+    # TFLite's object detection APIs require normalized anchors.
+    height, width = config['input_image_size']
+    normalize_factor = tf.constant(
+        [height, width, height, width], dtype=tf.float32
+    )
+    anchors = anchors / normalize_factor
 
   # There is no TF equivalent for TFLite's custom post-processing op.
   # So we add an 'empty' composite function here, that is legalized to the

official/vision/modeling/layers/detection_generator_test.py

@@ -150,6 +150,7 @@ def testDetectionsOutputShape(self, nms_version, has_att_heads, use_cpu_nms,
         'nms_score_threshold': 0.01,
         'nms_iou_threshold': 0.5,
         'input_image_size': [224, 224],
+        'normalize_anchor_coordinates': True,
     }
     kwargs = {
         'apply_nms': True,