tf-models-nightly 2.14.0.dev20231113__py2.py3-none-any.whl → 2.14.0.dev20231114__py2.py3-none-any.whl

official/projects/yolo/dataloaders/classification_input.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 
 """Classification decoder and parser."""
+from typing import List
+
 import tensorflow as tf, tf_keras
 from official.vision.dataloaders import classification_input
 from official.vision.ops import preprocess_ops
@@ -90,3 +92,19 @@ class Parser(classification_input.Parser):
     image = tf.image.convert_image_dtype(image, self._dtype)
     image = image / 255.0
     return image
+
+  @classmethod
+  def inference_fn(
+      cls, image: tf.Tensor, input_image_size: List[int], num_channels: int = 3
+  ) -> tf.Tensor:
+    """Builds image model inputs for serving."""
+
+    image = tf.cast(image, dtype=tf.float32)
+    image = preprocess_ops.center_crop_image(image)
+    image = tf.image.resize(
+        image, input_image_size, method=tf.image.ResizeMethod.BILINEAR
+    )
+
+    image.set_shape(input_image_size + [num_channels])
+    image = image / 255.0
+    return image

official/projects/yolo/serving/export_module_factory.py

@@ -23,11 +23,11 @@ from official.core import export_base
 from official.projects.yolo.configs import darknet_classification
 from official.projects.yolo.configs import yolo
 from official.projects.yolo.configs import yolov7
+from official.projects.yolo.dataloaders import classification_input
 from official.projects.yolo.modeling import factory as yolo_factory
 from official.projects.yolo.modeling.backbones import darknet  # pylint: disable=unused-import
 from official.projects.yolo.modeling.decoders import yolo_decoder  # pylint: disable=unused-import
 from official.projects.yolo.serving import model_fn as yolo_model_fn
-from official.vision.dataloaders import classification_input
 from official.vision.modeling import factory
 from official.vision.serving import export_utils
 

{tf_models_nightly-2.14.0.dev20231113.dist-info → tf_models_nightly-2.14.0.dev20231114.dist-info}/METADATA

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

{tf_models_nightly-2.14.0.dev20231113.dist-info → tf_models_nightly-2.14.0.dev20231114.dist-info}/RECORD

@@ -775,7 +775,7 @@ official/projects/yolo/configs/decoders.py,sha256=NRtd96kfDWgyxzZtd_zvbJ-UZWC_ra
 official/projects/yolo/configs/yolo.py,sha256=S3Ct_MQOeMvs5PqE6_RSizk6BRSVuQbMq4rwdOqX1CY,18648
 official/projects/yolo/configs/yolov7.py,sha256=ZjoMQSeBdESO8QirKSW9n4q0Is3toah0jTxlTwVAprY,12932
 official/projects/yolo/dataloaders/__init__.py,sha256=wrNoNRP6wGQBFDLVCYB01k1HgwOGlQ8iZnfqimxlDyk,610
-official/projects/yolo/dataloaders/classification_input.py,sha256=SJdhTMawIQX1MiJwYC6ZyiiAZjaU1lKcsw3YL093b88,3289
+official/projects/yolo/dataloaders/classification_input.py,sha256=LrSjUFywNn4r2q9BDg3-mg5QQ2uU0Tud7bDBsPTxw08,3800
 official/projects/yolo/dataloaders/tf_example_decoder.py,sha256=NFmXcEJEHB9y_pr6VHNupwzXkhZnaWBqaf7Raypt-9E,4820
 official/projects/yolo/dataloaders/yolo_input.py,sha256=1Jhw5pOOdonZyMlEA2JzCoWJ2g6MhZ4vEN3NG4_uaig,14616
 official/projects/yolo/losses/__init__.py,sha256=1ToRMjre4mErL4Ek4_dMVxMjXNPossNXggV8fqbISao,609
@@ -827,7 +827,7 @@ official/projects/yolo/optimization/configs/__init__.py,sha256=1ToRMjre4mErL4Ek4
 official/projects/yolo/optimization/configs/optimization_config.py,sha256=kl0F8qjqtgNbeuDH4mFxwna0Unm6bl49a6dfE4Z_s8A,2084
 official/projects/yolo/optimization/configs/optimizer_config.py,sha256=F5qE0kaUxGk9dW5R_I3Ofp-Myd_T3i1-p1eJ58-O6KM,2283
 official/projects/yolo/serving/__init__.py,sha256=1ToRMjre4mErL4Ek4_dMVxMjXNPossNXggV8fqbISao,609
-official/projects/yolo/serving/export_module_factory.py,sha256=fGoMMnJvBT8h4XpI2uaoArk0BhoD6ecdW5fTHlggVNw,9948
+official/projects/yolo/serving/export_module_factory.py,sha256=jx0LyHEaUNDv9IkjUN_uY1XTwhc5UlImWLuk5V0YkLw,9955
 official/projects/yolo/serving/export_saved_model.py,sha256=wFjN6wAJ-ckNEtSTwMY6Kj4jDz7CSAytOqFy1AVCt7w,4584
 official/projects/yolo/serving/export_tflite.py,sha256=Pt1rkKu9TxVYB8HtLadJQ_buWzNXqWL8Zp0TFDPUXzw,902
 official/projects/yolo/serving/model_fn.py,sha256=_T28osfXUtnBU9GqULbJ6HMd44pJMRWYZfitjrhWnlo,3018
@@ -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.14.0.dev20231113.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.14.0.dev20231113.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.14.0.dev20231113.dist-info/METADATA,sha256=oTGr0oKVZFxOli0miW_tl8TpYOkw0X4lgwjChxdTbb8,1414
-tf_models_nightly-2.14.0.dev20231113.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.14.0.dev20231113.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.14.0.dev20231113.dist-info/RECORD,,
+tf_models_nightly-2.14.0.dev20231114.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.14.0.dev20231114.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.14.0.dev20231114.dist-info/METADATA,sha256=iapp0upkgITxQsnBFx0Rfb0AYSu-OVgytPfKrv6Etdw,1414
+tf_models_nightly-2.14.0.dev20231114.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.14.0.dev20231114.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.14.0.dev20231114.dist-info/RECORD,,