tf-models-nightly 2.15.0.dev20231129__py2.py3-none-any.whl → 2.15.0.dev20231130__py2.py3-none-any.whl

official/vision/configs/common.py

@@ -157,3 +157,11 @@ class TFLitePostProcessingConfig(hyperparams.Config):
   # Whether to omit the final nms placeholder op. If set to True, the output
   # will be a tuple of boxes, scores result right before the NMS operation.
   omit_nms: Optional[bool] = False
+  # The number of detections per class when using regular NMS.
+  detections_per_class: Optional[int] = 5
+  # Box scaling factors. It should agree with `box_coder_weights` defined in
+  # `DetectionGenerator`, which is in the format of [y, x, w, h].
+  y_scale: float = 1.0
+  x_scale: float = 1.0
+  w_scale: float = 1.0
+  h_scale: float = 1.0

official/vision/modeling/layers/detection_generator.py

@@ -764,7 +764,6 @@ def _generate_detections_tflite_implements_signature(
   Returns:
     An `experimental_implements` signature string.
   """
-  scale_value = 1.0
 
   implements_signature = [
       'name: "%s"' % 'TFLite_Detection_PostProcess',
@@ -772,16 +771,18 @@ def _generate_detections_tflite_implements_signature(
       % config['max_detections'],
       'attr { key: "max_classes_per_detection" value { i: %d } }'
       % config['max_classes_per_detection'],
+      'attr { key: "detections_per_class" value { i: %d } }'
+      % config.get('detections_per_class', 5),
       'attr { key: "use_regular_nms" value { b: %s } }'
       % str(config['use_regular_nms']).lower(),
       'attr { key: "nms_score_threshold" value { f: %f } }'
       % config['nms_score_threshold'],
       'attr { key: "nms_iou_threshold" value { f: %f } }'
       % config['nms_iou_threshold'],
-      'attr { key: "y_scale" value { f: %f } }' % scale_value,
-      'attr { key: "x_scale" value { f: %f } }' % scale_value,
-      'attr { key: "h_scale" value { f: %f } }' % scale_value,
-      'attr { key: "w_scale" value { f: %f } }' % scale_value,
+      'attr { key: "y_scale" value { f: %f } }' % config.get('y_scale', 1.0),
+      'attr { key: "x_scale" value { f: %f } }' % config.get('x_scale', 1.0),
+      'attr { key: "h_scale" value { f: %f } }' % config.get('h_scale', 1.0),
+      'attr { key: "w_scale" value { f: %f } }' % config.get('w_scale', 1.0),
       'attr { key: "num_classes" value { i: %d } }' % config['num_classes'],
   ]
   implements_signature = ' '.join(implements_signature)

{tf_models_nightly-2.15.0.dev20231129.dist-info → tf_models_nightly-2.15.0.dev20231130.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tf-models-nightly
-Version: 2.15.0.dev20231129
+Version: 2.15.0.dev20231130
 Summary: TensorFlow Official Models
 Home-page: https://github.com/tensorflow/models
 Author: Google Inc.

{tf_models_nightly-2.15.0.dev20231129.dist-info → tf_models_nightly-2.15.0.dev20231130.dist-info}/RECORD

@@ -912,7 +912,7 @@ official/vision/train_spatial_partitioning.py,sha256=LMr8ztJ9WF7yCkXB20AsS2B3G3o
 official/vision/configs/__init__.py,sha256=5Uk9jeIYHTEHXs8kBrq-vMa7cgg_JpLgSYuVaLPyUt8,1045
 official/vision/configs/backbones.py,sha256=K8ph1ORFKtOmHW_WA9xD3tjYKIEtRkzNJ2klhWFUogQ,5547
 official/vision/configs/backbones_3d.py,sha256=f1v1TMsBofFbWHGqFMcJJnopNum6Ua-FiWGlwhVcbUo,3614
-official/vision/configs/common.py,sha256=EmQ7CpbmdK26LsRDiHY0FMbgabcIXU2EWV6Ga7Qx1NA,5014
+official/vision/configs/common.py,sha256=8DjWK7LsVtdYxAYm-mxspj9PabEGkVCDgmC7dzY3ks0,5354
 official/vision/configs/decoders.py,sha256=5gmmmvrh0hxfxFKsqstOSBLtJN_lk8jbejCjefV0J-U,2080
 official/vision/configs/image_classification.py,sha256=0fUVS0EI6aSC4fU_YDniiiO6O4ooylRVjKkOMpqisyA,24563
 official/vision/configs/image_classification_test.py,sha256=dH4MDaplTEcon4osucrblkx06WD0Vh-YQ5249wQYoM4,1867
@@ -1039,7 +1039,7 @@ official/vision/modeling/layers/__init__.py,sha256=Yk0zkJVVgwi--tv5-NKIAKHCzpFw9
 official/vision/modeling/layers/box_sampler.py,sha256=a7ssbMDR5OEoSIGmRU3HJGoWYNXJA6EmCg5T1K36wco,3411
 official/vision/modeling/layers/deeplab.py,sha256=YBRa3uO_rqzxr6dS9beUZexb0hDE_eJkWC9l15y6OHI,8571
 official/vision/modeling/layers/deeplab_test.py,sha256=Vh8DHsvtUVQ6pzOSvY9QtPz1pJmq7zlDzxzFZpuNUOM,1904
-official/vision/modeling/layers/detection_generator.py,sha256=AtW01DnCJfqG79mgCRjSP7llb-AifgJbV5x6O5riqT4,65453
+official/vision/modeling/layers/detection_generator.py,sha256=SRe9rBM_IqIbg9zczk9eyTbFJXAGUQXz3vOli6YyI5A,65601
 official/vision/modeling/layers/detection_generator_test.py,sha256=9Se5WEizWUwbSJc5Cf5MU7qlMA3KXDyW4WCZfgPHMso,13577
 official/vision/modeling/layers/edgetpu.py,sha256=ofhZtwYq0qjU7NcQcaXIYSKvEG-PZPL7p0U1g3eXEPA,16737
 official/vision/modeling/layers/edgetpu_test.py,sha256=oY2czEzDQq40IlzcaWazw89mtPsDx4K5CCTyesdsksw,10120
@@ -1158,9 +1158,9 @@ tensorflow_models/__init__.py,sha256=Ciz_YBke6teb6y42QyQTUBDdXJAiV7Qdu1zOoZvYiKw
 tensorflow_models/tensorflow_models_test.py,sha256=qUBLFZg7rmKkVQ3cHJVlkoid8cPqjjyc2ZiWtjQO5_o,1395
 tensorflow_models/nlp/__init__.py,sha256=3dULDpUBpDi9vljpXadq6oJrWH4y6z42Bz2d3hopYZw,807
 tensorflow_models/vision/__init__.py,sha256=4y77XkHaH8qLls3-6ta4tMp3Xj8CLbB0ihH91HsQ9z4,833
-tf_models_nightly-2.15.0.dev20231129.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.15.0.dev20231129.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.15.0.dev20231129.dist-info/METADATA,sha256=6GDquWhKpS2xlX4z3MzKfOQpRKR4SJiRpOkDKOEicsA,1414
-tf_models_nightly-2.15.0.dev20231129.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.15.0.dev20231129.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.15.0.dev20231129.dist-info/RECORD,,
+tf_models_nightly-2.15.0.dev20231130.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.15.0.dev20231130.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.15.0.dev20231130.dist-info/METADATA,sha256=-Nppc2kRx3c32vfT8uwdNaR6Q2R92fJqftbHoQ1p8sg,1414
+tf_models_nightly-2.15.0.dev20231130.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.15.0.dev20231130.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.15.0.dev20231130.dist-info/RECORD,,