Internal change

tensorflower-gardener · fyangf · commit a3fc65689fc4 · 2023-03-20T21:36:26.000-07:00
PiperOrigin-RevId: 505244177
diff --git a/official/vision/configs/retinanet.py b/official/vision/configs/retinanet.py
@@ -135,6 +135,8 @@ class DetectionGenerator(hyperparams.Config):
   )
   # Return decoded boxes/scores even if apply_nms is set `True`.
   return_decoded: Optional[bool] = None
+  # Only works when nms_version='v2'.
+  use_class_agnostic_nms: Optional[bool] = False
 
 
 @dataclasses.dataclass
diff --git a/official/vision/modeling/factory.py b/official/vision/modeling/factory.py
@@ -323,6 +323,7 @@ def build_retinanet(
       soft_nms_sigma=generator_config.soft_nms_sigma,
       tflite_post_processing_config=tflite_post_processing_config,
       return_decoded=generator_config.return_decoded,
+      use_class_agnostic_nms=generator_config.use_class_agnostic_nms,
   )
 
   model = retinanet_model.RetinaNetModel(

Original file line number	Diff line number	Diff line change
`@@ -135,6 +135,8 @@ class DetectionGenerator(hyperparams.Config):`
`135`	`135`	`)`
`136`	`136`	# Return decoded boxes/scores even if apply_nms is set `True`.
`137`	`137`	`return_decoded: Optional[bool] = None`
	`138`	`+ # Only works when nms_version='v2'.`
	`139`	`+ use_class_agnostic_nms: Optional[bool] = False`
`138`	`140`
`139`	`141`
`140`	`142`	`@dataclasses.dataclass`
Original file line number	Diff line number	Diff line change
`@@ -323,6 +323,7 @@ def build_retinanet(`
`323`	`323`	`soft_nms_sigma=generator_config.soft_nms_sigma,`
`324`	`324`	`tflite_post_processing_config=tflite_post_processing_config,`
`325`	`325`	`return_decoded=generator_config.return_decoded,`
	`326`	`+ use_class_agnostic_nms=generator_config.use_class_agnostic_nms,`
`326`	`327`	`)`
`327`	`328`
`328`	`329`	`model = retinanet_model.RetinaNetModel(`