Support output decoded boxes (before NMS) even when NMS is applied.

PiperOrigin-RevId: 501150589

Author: tensorflower-gardener

official/vision/configs/retinanet.py

@@ -126,6 +126,8 @@ class DetectionGenerator(hyperparams.Config):
   # custom NMS op and override above parameters.
   tflite_post_processing: common.TFLitePostProcessingConfig = common.TFLitePostProcessingConfig(
   )
+  # Return decoded boxes/scores even if apply_nms is set `True`.
+  return_decoded: Optional[bool] = None
 
 
 @dataclasses.dataclass

official/vision/modeling/factory.py

@@ -321,7 +321,9 @@ def build_retinanet(
       nms_version=generator_config.nms_version,
       use_cpu_nms=generator_config.use_cpu_nms,
       soft_nms_sigma=generator_config.soft_nms_sigma,
-      tflite_post_processing_config=tflite_post_processing_config)
+      tflite_post_processing_config=tflite_post_processing_config,
+      return_decoded=generator_config.return_decoded,
+  )
 
   model = retinanet_model.RetinaNetModel(
       backbone,