tf-models-nightly 2.12.0.dev20230409__py2.py3-none-any.whl → 2.12.0.dev20230411__py2.py3-none-any.whl

official/vision/ops/augment.py

@@ -31,7 +31,6 @@ import inspect
 import math
 from typing import Any, List, Iterable, Optional, Text, Tuple
 
-from keras.layers.preprocessing import image_preprocessing as image_ops
 import numpy as np
 import tensorflow as tf
 from tensorflow_addons import image as tfa_image
@@ -171,15 +170,96 @@ def _convert_angles_to_transform(angles: tf.Tensor, image_width: tf.Tensor,
   )
 
 
+def _apply_transform_to_images(
+    images,
+    transforms,
+    fill_mode='reflect',
+    fill_value=0.0,
+    interpolation='bilinear',
+    output_shape=None,
+    name=None,
+):
+  """Applies the given transform(s) to the image(s).
+
+  Args:
+    images: A tensor of shape `(num_images, num_rows, num_columns,
+      num_channels)` (NHWC). The rank must be statically known (the shape is
+      not `TensorShape(None)`).
+    transforms: Projective transform matrix/matrices. A vector of length 8 or
+      tensor of size N x 8. If one row of transforms is [a0, a1, a2, b0, b1,
+      b2, c0, c1], then it maps the *output* point `(x, y)` to a transformed
+      *input* point `(x', y') = ((a0 x + a1 y + a2) / k, (b0 x + b1 y + b2) /
+      k)`, where `k = c0 x + c1 y + 1`. The transforms are *inverted* compared
+      to the transform mapping input points to output points. Note that
+      gradients are not backpropagated into transformation parameters.
+    fill_mode: Points outside the boundaries of the input are filled according
+      to the given mode (one of `{"constant", "reflect", "wrap", "nearest"}`).
+    fill_value: a float represents the value to be filled outside the
+      boundaries when `fill_mode="constant"`.
+    interpolation: Interpolation mode. Supported values: `"nearest"`,
+      `"bilinear"`.
+    output_shape: Output dimension after the transform, `[height, width]`. If
+      `None`, output is the same size as input image.
+    name: The name of the op.  Fill mode behavior for each valid value is as
+      follows
+      - `"reflect"`: `(d c b a | a b c d | d c b a)` The input is extended by
+      reflecting about the edge of the last pixel.
+      - `"constant"`: `(k k k k | a b c d | k k k k)` The input is extended by
+      filling all values beyond the edge with the same constant value k = 0.
+      - `"wrap"`: `(a b c d | a b c d | a b c d)` The input is extended by
+      wrapping around to the opposite edge.
+      - `"nearest"`: `(a a a a | a b c d | d d d d)` The input is extended by
+      the nearest pixel.  Input shape: 4D tensor with shape:
+      `(samples, height, width, channels)`, in `"channels_last"` format.
+      Output shape: 4D tensor with shape: `(samples, height, width, channels)`,
+      in `"channels_last"` format.
+
+  Returns:
+    Image(s) with the same type and shape as `images`, with the given
+    transform(s) applied. Transformed coordinates outside of the input image
+    will be filled with zeros.
+  """
+  with tf.name_scope(name or 'transform'):
+    if output_shape is None:
+      output_shape = tf.shape(images)[1:3]
+      if not tf.executing_eagerly():
+        output_shape_value = tf.get_static_value(output_shape)
+        if output_shape_value is not None:
+          output_shape = output_shape_value
+
+    output_shape = tf.convert_to_tensor(
+        output_shape, tf.int32, name='output_shape'
+    )
+
+    if not output_shape.get_shape().is_compatible_with([2]):
+      raise ValueError(
+          'output_shape must be a 1-D Tensor of 2 elements: '
+          'new_height, new_width, instead got '
+          f'output_shape={output_shape}'
+      )
+
+    fill_value = tf.convert_to_tensor(fill_value, tf.float32, name='fill_value')
+
+    return tf.raw_ops.ImageProjectiveTransformV3(
+        images=images,
+        output_shape=output_shape,
+        fill_value=fill_value,
+        transforms=transforms,
+        fill_mode=fill_mode.upper(),
+        interpolation=interpolation.upper(),
+    )
+
+
 def transform(image: tf.Tensor, transforms) -> tf.Tensor:
-  """Prepares input data for `image_ops.transform`."""
+  """Transforms an image."""
   original_ndims = tf.rank(image)
   transforms = tf.convert_to_tensor(transforms, dtype=tf.float32)
   if transforms.shape.rank == 1:
     transforms = transforms[None]
   image = to_4d(image)
-  image = image_ops.transform(
-      images=image, transforms=transforms, interpolation='nearest')
+  image = _apply_transform_to_images(
+      images=image, transforms=transforms, interpolation='nearest'
+  )
   return from_4d(image, original_ndims)
 
 

{tf_models_nightly-2.12.0.dev20230409.dist-info → tf_models_nightly-2.12.0.dev20230411.dist-info}/METADATA

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

{tf_models_nightly-2.12.0.dev20230409.dist-info → tf_models_nightly-2.12.0.dev20230411.dist-info}/RECORD

@@ -973,7 +973,7 @@ official/vision/ops/anchor.py,sha256=jW8F1js-bFCo4JGP4rFw4ti6mHhUuEfTEkQ1mvLhfe8
 official/vision/ops/anchor_generator.py,sha256=WwASWaE5uR9oIB69lnK8mW2xUfRnLY0LsgyVX1BwsQQ,7234
 official/vision/ops/anchor_generator_test.py,sha256=aihKzRV8SQFhNSzS9_SpJNSadL4KcUcj0vT5dgmnkIQ,5286
 official/vision/ops/anchor_test.py,sha256=UHue1kHD184Lh5eJzWQ2w0UAZk9FjolYJ9Eg6b1hIbo,7623
-official/vision/ops/augment.py,sha256=fPoiF88kWHSL73se_PUFgEQtzAhqUsGpmwWwi-tOr9E,91930
+official/vision/ops/augment.py,sha256=-8Ag70G1lYw4EEvIgitbX8er9WgPYhlc-QNC4N5UK20,95292
 official/vision/ops/augment_test.py,sha256=r7oM3uUgmSnwPhQGiamkwt9s3wmbavykGEiC7Cj4Mho,19453
 official/vision/ops/box_matcher.py,sha256=DHx5Y9Le4CiIK73oxk1DBjB28-SRuo-IUEd3BzdrTa4,9057
 official/vision/ops/box_matcher_test.py,sha256=1QPcFSKhKCx2L65tH7suEEosCb2oGLLF1xPmHSmGT5E,2428
@@ -1071,9 +1071,9 @@ tensorflow_models/__init__.py,sha256=Ciz_YBke6teb6y42QyQTUBDdXJAiV7Qdu1zOoZvYiKw
 tensorflow_models/tensorflow_models_test.py,sha256=Kz2y4V-rtBhZFFfKD2soCq52hviSfJVV1L2ztqS-9oM,1385
 tensorflow_models/nlp/__init__.py,sha256=3dULDpUBpDi9vljpXadq6oJrWH4y6z42Bz2d3hopYZw,807
 tensorflow_models/vision/__init__.py,sha256=4y77XkHaH8qLls3-6ta4tMp3Xj8CLbB0ihH91HsQ9z4,833
-tf_models_nightly-2.12.0.dev20230409.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.12.0.dev20230409.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.12.0.dev20230409.dist-info/METADATA,sha256=Zv-kbAY3QAz-3OomoFuLEY9egM0dJcLZywlkAtKwJQs,1426
-tf_models_nightly-2.12.0.dev20230409.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.12.0.dev20230409.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.12.0.dev20230409.dist-info/RECORD,,
+tf_models_nightly-2.12.0.dev20230411.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.12.0.dev20230411.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.12.0.dev20230411.dist-info/METADATA,sha256=UqRWh_MwOd_8qU5NYSZ-8lAym0CDeCCWHPoAKiMKpwU,1426
+tf_models_nightly-2.12.0.dev20230411.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.12.0.dev20230411.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.12.0.dev20230411.dist-info/RECORD,,