tf-models-nightly 2.17.0.dev20240424__py2.py3-none-any.whl → 2.17.0.dev20240425__py2.py3-none-any.whl

official/vision/ops/preprocess_ops.py

@@ -43,7 +43,7 @@ def clip_or_pad_to_fixed_size(input_tensor, size, constant_values=0):
   Args:
     input_tensor: `Tensor` with any dimension.
     size: `int` number for the first dimension of output Tensor.
-    constant_values: `int` value assigned to the paddings.
+    constant_values: `int` or `str` value assigned to the paddings.
 
   Returns:
     `Tensor` with the first dimension padded to `size`.
@@ -66,7 +66,7 @@ def clip_or_pad_to_fixed_size(input_tensor, size, constant_values=0):
 
   # Pads input tensor to the fixed first dimension.
   paddings = tf.cast(
-      constant_values * tf.ones(padding_shape), input_tensor.dtype
+      tf.fill(dims=padding_shape, value=constant_values), input_tensor.dtype
   )
   padded_tensor = tf.concat([input_tensor, paddings], axis=0)
   output_shape = input_shape

official/vision/ops/preprocess_ops_test.py

@@ -36,30 +36,34 @@ def _encode_image(image_array, fmt):
 class InputUtilsTest(parameterized.TestCase, tf.test.TestCase):
 
   @parameterized.parameters(
-      ([1], 10),
-      ([1, 2], 10),
-      ([1, 2, 3], 10),
-      ([11], 10),
-      ([12, 2], 10),
-      ([13, 2, 3], 10),
+      ([1], 10, 1.0, 0.0),
+      ([1, 2], 10, 1.0, 0.0),
+      ([1, 2, 3], 10, 1.0, 0.0),
+      ([11], 10, 1.0, 0.0),
+      ([12, 2], 10, 1.0, 0.0),
+      ([13, 2, 3], 10, 1.0, 0.0),
+      ([1, 2], 10, 'test', 'pad'),
   )
-  def test_pad_to_fixed_size(self, input_shape, output_size):
+  def test_pad_to_fixed_size(
+      self, input_shape, output_size, original_val, pad_val
+  ):
     # Copies input shape to padding shape.
     clip_shape = input_shape[:]
     clip_shape[0] = min(output_size, clip_shape[0])
     padding_shape = input_shape[:]
     padding_shape[0] = max(output_size - input_shape[0], 0)
     expected_outputs = np.concatenate(
-        [np.ones(clip_shape), np.zeros(padding_shape)], axis=0
+        [np.full(clip_shape, original_val), np.full(padding_shape, pad_val)],
+        axis=0,
     )
 
-    data = tf.ones(input_shape)
+    data = tf.fill(input_shape, original_val)
     output_data = preprocess_ops.clip_or_pad_to_fixed_size(
-        data, output_size, constant_values=0
+        data, output_size, constant_values=pad_val
     )
     output_data = output_data.numpy()
-    self.assertAllClose(output_size, output_data.shape[0])
-    self.assertAllClose(expected_outputs, output_data)
+    self.assertAllEqual(output_size, output_data.shape[0])
+    self.assertAllEqual(expected_outputs, output_data)
 
   @parameterized.named_parameters(
       dict(

{tf_models_nightly-2.17.0.dev20240424.dist-info → tf_models_nightly-2.17.0.dev20240425.dist-info}/METADATA

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

{tf_models_nightly-2.17.0.dev20240424.dist-info → tf_models_nightly-2.17.0.dev20240425.dist-info}/RECORD

@@ -1117,10 +1117,10 @@ official/vision/ops/iou_similarity_test.py,sha256=x5jlcMqMCUYC5cRgdbR0VlAW67AoXo
 official/vision/ops/mask_ops.py,sha256=cZLpIowzEA57bXPDbVXa6mktZVHvGSH-TQ1CxHjpQXw,10270
 official/vision/ops/mask_ops_test.py,sha256=D3xbbbleJd4HkpWOSDSEy6hNihsRBY93BqPF6JP-dJk,2835
 official/vision/ops/nms.py,sha256=bKYDAtyV5j6PG7g-RGF2ZccCI5V1xVvuajNblCy1TGs,8125
-official/vision/ops/preprocess_ops.py,sha256=tDW9apUocwSzGnKkgSrKGwqbcZpb6-WQ8nGnW1_ds20,42569
+official/vision/ops/preprocess_ops.py,sha256=JnelXrBbfFkdyz5UbrZxnD_N4RdN_axFa7FNFxeSpdQ,42588
 official/vision/ops/preprocess_ops_3d.py,sha256=o0ivTBK2BwaiPp_NhVG_yBc10VUySxfE7eKQkL7RNaU,16762
 official/vision/ops/preprocess_ops_3d_test.py,sha256=LA3-Ue4FTjdsr5Kl_BnpAMNcfikWylMisD2GrBTOLzA,9309
-official/vision/ops/preprocess_ops_test.py,sha256=FY3EUvQIb82fYqYJPmzkE7pmkhXIQrd7JNLGFPB6SXI,17213
+official/vision/ops/preprocess_ops_test.py,sha256=VXcILiqjWPcVHCVjx7i0iiPV6FP0s9pZ7LhXPwVPNHs,17392
 official/vision/ops/sampling_ops.py,sha256=1jywCA_E4qDUFWsykCLUwZsWtQSR0QREXdJhvP5qCvc,16072
 official/vision/ops/spatial_transform_ops.py,sha256=PVEJGAn0ygtsrid84vD5GgV0jsjyWoNn14RBzreMxM4,38389
 official/vision/ops/target_gather.py,sha256=Ir3X76yXYEVFSYX5h-yfS8SMkY37GYuypBP2C8ykggo,3965
@@ -1206,9 +1206,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=nc6A9K53OGqF25xN5St8EiWvdVbda
 tensorflow_models/nlp/__init__.py,sha256=4tA5Pf4qaFwT-fIFOpX7x7FHJpnyJT-5UgOeFYTyMlc,807
 tensorflow_models/uplift/__init__.py,sha256=mqfa55gweOdpKoaQyid4A_4u7xw__FcQeSIF0k_pYmI,999
 tensorflow_models/vision/__init__.py,sha256=zBorY_v5xva1uI-qxhZO3Qh-Dii-Suq6wEYh6hKHDfc,833
-tf_models_nightly-2.17.0.dev20240424.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.17.0.dev20240424.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.17.0.dev20240424.dist-info/METADATA,sha256=emMHUD-L2fms_4usmoJBBp54R0EiQttfunbwANOvPfk,1432
-tf_models_nightly-2.17.0.dev20240424.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.17.0.dev20240424.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.17.0.dev20240424.dist-info/RECORD,,
+tf_models_nightly-2.17.0.dev20240425.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.17.0.dev20240425.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.17.0.dev20240425.dist-info/METADATA,sha256=vMmHziifUoAFoNJrYv0PNCkmAv4KkUw57jKbI2FqaHg,1432
+tf_models_nightly-2.17.0.dev20240425.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.17.0.dev20240425.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.17.0.dev20240425.dist-info/RECORD,,