No public description

PiperOrigin-RevId: 553646554

No public description
PiperOrigin-RevId: 553646554
c4cc4ba5 · Fan Yang · A. Unique TensorFlower · 37f09530 · c4cc4ba5
隐藏空白更改
内联并排

Showing with 6 addition and 3 deletion

official/vision/modeling/factory.py official/vision/modeling/factory.py +6 -3

未找到文件。
--- a/official/vision/modeling/factory.py
+++ b/official/vision/modeling/factory.py
@@ -311,10 +311,13 @@ def build_retinanet(
    _ = head(decoder_features)

  # Add `input_image_size` into `tflite_post_processing_config`.
-  tflite_post_processing_config = generator_config.tflite_post_processing.as_dict(
+  tflite_post_processing_config = (
+      generator_config.tflite_post_processing.as_dict()
+  )
+  tflite_post_processing_config['input_image_size'] = (
+      input_specs.shape[1],
+      input_specs.shape[2],
  )
-  tflite_post_processing_config['input_image_size'] = (input_specs.shape[1],
-                                                       input_specs.shape[2])
  detection_generator_obj = detection_generator.MultilevelDetectionGenerator(
      apply_nms=generator_config.apply_nms,
      pre_nms_top_k=generator_config.pre_nms_top_k,