tf-models-nightly 2.20.0.dev20250825__py2.py3-none-any.whl → 2.20.0.dev20250826__py2.py3-none-any.whl

official/vision/ops/preprocess_ops.py

@@ -795,6 +795,20 @@ def horizontal_flip_masks(masks):
     return masks[:, :, ::-1]
 
 
+def vertical_flip_image(image):
+  """Flips image vertically."""
+  return tf.image.flip_up_down(image)
+
+
+def vertical_flip_masks(masks):
+  """Flips masks vertically. Expects rank-3 input dimensions."""
+  # For masks shape of [h, w, 1].
+  if masks.shape[-1] == 1:
+    return masks[::-1, :, :]
+  else:
+    return masks[:, ::-1, :]
+
+
 def random_horizontal_flip(
     image, normalized_boxes=None, masks=None, seed=1, prob=0.5
 ):
@@ -900,7 +914,7 @@ def random_vertical_flip(
     do_flip = tf.less(tf.random.uniform([], seed=seed), prob)
 
     image = tf.cond(
-        do_flip, lambda: tf.image.flip_up_down(image), lambda: image
+        do_flip, lambda: vertical_flip_image(image), lambda: image
     )
 
     if normalized_boxes is not None:
@@ -913,7 +927,7 @@ def random_vertical_flip(
     if masks is not None:
       masks = tf.cond(
           do_flip,
-          lambda: tf.image.flip_up_down(masks[..., None])[..., 0],
+          lambda: vertical_flip_masks(masks),
           lambda: masks,
       )
 

{tf_models_nightly-2.20.0.dev20250825.dist-info → tf_models_nightly-2.20.0.dev20250826.dist-info}/METADATA

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

{tf_models_nightly-2.20.0.dev20250825.dist-info → tf_models_nightly-2.20.0.dev20250826.dist-info}/RECORD

@@ -1159,7 +1159,7 @@ official/vision/ops/iou_similarity_test.py,sha256=4etrxHdZzxOZ6ZwGMGbBR8WXsUP-CW
 official/vision/ops/mask_ops.py,sha256=K_8mjLgH6A4wduZAPA73egfgbsnV9SBTXy6cH1J_l2c,10250
 official/vision/ops/mask_ops_test.py,sha256=0oExjZHHLTxagyFBadyoLLuQfNlr18A682WXZ-NhQUM,2816
 official/vision/ops/nms.py,sha256=JMKQdPU53zlxhMaj2quxJPy2wyFAbzzDRmHyM8oRF8U,8106
-official/vision/ops/preprocess_ops.py,sha256=p0C38b2_uEAuCr1aRZ0ju81BXxOEMDmELWhE8mMaKHg,43301
+official/vision/ops/preprocess_ops.py,sha256=fTafI8_HiciUIdX-UGLpJnV0rjOfZ1h2776ZwY3MBmI,43608
 official/vision/ops/preprocess_ops_3d.py,sha256=D9ojriFAJg3yj3Od4MmXDy_kIlt_csUvq6kOt99jPmE,16762
 official/vision/ops/preprocess_ops_3d_test.py,sha256=tSXBFtt1eRSAJnyLI0Lr7H7XHRozDjrSJWrvBug84H0,9309
 official/vision/ops/preprocess_ops_test.py,sha256=37y-P2IDfRUOvZeA5aktgfjxxR9DaqXLsQJzGQaKz-g,17638
@@ -1248,9 +1248,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=yiAneltAW3NHSj3fUSvHNBjfq0MGZ
 tensorflow_models/nlp/__init__.py,sha256=8uQd4wI6Zc4IJMPjtQifMeWVbPFkTxqYh66wfivCOL4,807
 tensorflow_models/uplift/__init__.py,sha256=NzaweFf4ZmhRb2l_fuV6bP-2N8oSO3xu6xJqVb1UmpY,999
 tensorflow_models/vision/__init__.py,sha256=ks420Ooqzi0hU7HnQpM5rylLaE-YcJdJkBx_umVaXlE,833
-tf_models_nightly-2.20.0.dev20250825.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.20.0.dev20250825.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.20.0.dev20250825.dist-info/METADATA,sha256=nRNh-Zm2wMloIQYole__lXx_QMlW9v-O7W8CPAE1uVY,1432
-tf_models_nightly-2.20.0.dev20250825.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.20.0.dev20250825.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.20.0.dev20250825.dist-info/RECORD,,
+tf_models_nightly-2.20.0.dev20250826.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.20.0.dev20250826.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.20.0.dev20250826.dist-info/METADATA,sha256=IXIag2CwjXlDY4MZK5_biOc-rwLsW_sn0epOTeUL_hU,1432
+tf_models_nightly-2.20.0.dev20250826.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.20.0.dev20250826.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.20.0.dev20250826.dist-info/RECORD,,