tf-models-nightly 2.18.0.dev20240919__py2.py3-none-any.whl → 2.18.0.dev20240920__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. official/vision/utils/object_detection/visualization_utils.py +5 -0
  2. {tf_models_nightly-2.18.0.dev20240919.dist-info → tf_models_nightly-2.18.0.dev20240920.dist-info}/METADATA +1 -1
  3. {tf_models_nightly-2.18.0.dev20240919.dist-info → tf_models_nightly-2.18.0.dev20240920.dist-info}/RECORD +7 -7
  4. {tf_models_nightly-2.18.0.dev20240919.dist-info → tf_models_nightly-2.18.0.dev20240920.dist-info}/AUTHORS +0 -0
  5. {tf_models_nightly-2.18.0.dev20240919.dist-info → tf_models_nightly-2.18.0.dev20240920.dist-info}/LICENSE +0 -0
  6. {tf_models_nightly-2.18.0.dev20240919.dist-info → tf_models_nightly-2.18.0.dev20240920.dist-info}/WHEEL +0 -0
  7. {tf_models_nightly-2.18.0.dev20240919.dist-info → tf_models_nightly-2.18.0.dev20240920.dist-info}/top_level.txt +0 -0
official/vision/utils/object_detection/visualization_utils.py CHANGED
@@ -438,6 +438,11 @@ def visualize_outputs(
438
438
  )
439
439
  return tf.cast(images, dtype=tf.uint8)
440
440
 
441
+ if images.shape[3] > 3:
442
+ images = images[:, :, :, 0:3]
443
+ elif images.shape[3] == 1:
444
+ images = tf.image.grayscale_to_rgb(images)
445
+
441
446
  images = tf.nest.map_structure(
442
447
  tf.identity,
443
448
  tf.map_fn(
{tf_models_nightly-2.18.0.dev20240919.dist-info → tf_models_nightly-2.18.0.dev20240920.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: tf-models-nightly
3
- Version: 2.18.0.dev20240919
3
+ Version: 2.18.0.dev20240920
4
4
  Summary: TensorFlow Official Models
5
5
  Home-page: https://github.com/tensorflow/models
6
6
  Author: Google Inc.
{tf_models_nightly-2.18.0.dev20240919.dist-info → tf_models_nightly-2.18.0.dev20240920.dist-info}/RECORD RENAMED
@@ -1187,7 +1187,7 @@ official/vision/utils/object_detection/preprocessor.py,sha256=mY9O3SAXLLd7juSDuK
1187
1187
  official/vision/utils/object_detection/region_similarity_calculator.py,sha256=oZCopHw-mUEwpUiqZbxZO0qY7Xc1eiXvo5Na68O6fbM,4554
1188
1188
  official/vision/utils/object_detection/shape_utils.py,sha256=kyAIT_qReNoRLnGHyD2EHPutbi2zd1NPwsHmA5v_9lo,3618
1189
1189
  official/vision/utils/object_detection/target_assigner.py,sha256=HNAwUhHQLo8ZTU-hTkvT1znypa8jgAhYAoX5veMTntk,24208
1190
- official/vision/utils/object_detection/visualization_utils.py,sha256=Hk13qz1xD2yf6021T2wiBEf5c3UVI7M13xSwf_8OmNI,40267
1190
+ official/vision/utils/object_detection/visualization_utils.py,sha256=NYQQtyuGvuRatKCxfRSh_cAonps337iv0Y9Wsh8pwdU,40404
1191
1191
  orbit/__init__.py,sha256=VSL7DNfTZmEG8eMdqhQOEp63wtnkQtRhI2Yp99eWWPg,1117
1192
1192
  orbit/controller.py,sha256=px_tjBiegR2U4NQoe8DeD7eUTvQxGrF4RnhMh93bITw,25368
1193
1193
  orbit/controller_test.py,sha256=Jn-yBUVxqeIwxiFTTWkBLAkV6_XaUKKqMJrvL5kKeMs,31802
@@ -1222,9 +1222,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=nc6A9K53OGqF25xN5St8EiWvdVbda
1222
1222
  tensorflow_models/nlp/__init__.py,sha256=4tA5Pf4qaFwT-fIFOpX7x7FHJpnyJT-5UgOeFYTyMlc,807
1223
1223
  tensorflow_models/uplift/__init__.py,sha256=mqfa55gweOdpKoaQyid4A_4u7xw__FcQeSIF0k_pYmI,999
1224
1224
  tensorflow_models/vision/__init__.py,sha256=zBorY_v5xva1uI-qxhZO3Qh-Dii-Suq6wEYh6hKHDfc,833
1225
- tf_models_nightly-2.18.0.dev20240919.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
1226
- tf_models_nightly-2.18.0.dev20240919.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
1227
- tf_models_nightly-2.18.0.dev20240919.dist-info/METADATA,sha256=z_bABqB1Cm6qU18Rku98ShkjheDYP-JbtDnzitB16cw,1432
1228
- tf_models_nightly-2.18.0.dev20240919.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
1229
- tf_models_nightly-2.18.0.dev20240919.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
1230
- tf_models_nightly-2.18.0.dev20240919.dist-info/RECORD,,
1225
+ tf_models_nightly-2.18.0.dev20240920.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
1226
+ tf_models_nightly-2.18.0.dev20240920.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
1227
+ tf_models_nightly-2.18.0.dev20240920.dist-info/METADATA,sha256=KoU9kHjiWil7ROKE1hAEfoJ2IHW6XIt9_Elsk1J_H4g,1432
1228
+ tf_models_nightly-2.18.0.dev20240920.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
1229
+ tf_models_nightly-2.18.0.dev20240920.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
1230
+ tf_models_nightly-2.18.0.dev20240920.dist-info/RECORD,,