tf-models-nightly 2.11.0.dev20230303__py2.py3-none-any.whl → 2.11.0.dev20230305__py2.py3-none-any.whl

official/legacy/detection/ops/nms.py

@@ -67,8 +67,9 @@ def _suppression_loop_body(boxes, iou_threshold, output_size, idx):
     output_size: the updated output_size.
     idx: the updated induction variable.
   """
-  num_tiles = tf.shape(boxes)[1] // NMS_TILE_SIZE
-  batch_size = tf.shape(boxes)[0]
+  boxes_shape = tf.shape(boxes)
+  num_tiles = boxes_shape[1] // NMS_TILE_SIZE
+  batch_size = boxes_shape[0]
 
   # Iterates over tiles that can possibly suppress the current tile.
   box_slice = tf.slice(boxes, [0, idx * NMS_TILE_SIZE, 0],
@@ -97,7 +98,7 @@ def _suppression_loop_body(boxes, iou_threshold, output_size, idx):
   boxes = tf.tile(tf.expand_dims(
       box_slice, [1]), [1, num_tiles, 1, 1]) * mask + tf.reshape(
           boxes, [batch_size, num_tiles, NMS_TILE_SIZE, 4]) * (1 - mask)
-  boxes = tf.reshape(boxes, [batch_size, -1, 4])
+  boxes = tf.reshape(boxes, boxes_shape)
 
   # Updates output_size.
   output_size += tf.reduce_sum(

official/vision/ops/nms.py

@@ -65,8 +65,9 @@ def _suppression_loop_body(boxes, iou_threshold, output_size, idx):
     output_size: the updated output_size.
     idx: the updated induction variable.
   """
-  num_tiles = tf.shape(boxes)[1] // NMS_TILE_SIZE
-  batch_size = tf.shape(boxes)[0]
+  boxes_shape = tf.shape(boxes)
+  num_tiles = boxes_shape[1] // NMS_TILE_SIZE
+  batch_size = boxes_shape[0]
 
   # Iterates over tiles that can possibly suppress the current tile.
   box_slice = tf.slice(boxes, [0, idx * NMS_TILE_SIZE, 0],
@@ -95,7 +96,7 @@ def _suppression_loop_body(boxes, iou_threshold, output_size, idx):
   boxes = tf.tile(tf.expand_dims(
       box_slice, [1]), [1, num_tiles, 1, 1]) * mask + tf.reshape(
           boxes, [batch_size, num_tiles, NMS_TILE_SIZE, 4]) * (1 - mask)
-  boxes = tf.reshape(boxes, [batch_size, -1, 4])
+  boxes = tf.reshape(boxes, boxes_shape)
 
   # Updates output_size.
   output_size += tf.reduce_sum(

{tf_models_nightly-2.11.0.dev20230303.dist-info → tf_models_nightly-2.11.0.dev20230305.dist-info}/METADATA

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

{tf_models_nightly-2.11.0.dev20230303.dist-info → tf_models_nightly-2.11.0.dev20230305.dist-info}/RECORD

@@ -99,7 +99,7 @@ official/legacy/detection/modeling/architecture/nn_ops.py,sha256=iWYYyZZ-I4t3m4i
 official/legacy/detection/modeling/architecture/resnet.py,sha256=zmBJrsC7pXLExPwtt03cnvYuaTlcInpphycjiIanndc,13159
 official/legacy/detection/modeling/architecture/spinenet.py,sha256=EiKw4tHOHc2fgGip-PQ9I5MD1ifMKGuv6TU51rKwGbY,17270
 official/legacy/detection/ops/__init__.py,sha256=XKqEvUISyqNK_cFqr7umxt6r-vnABJ2OqGEKE395w20,609
-official/legacy/detection/ops/nms.py,sha256=99rwdOtCeasTRM0NWK84COnEacLZZrXBHWvwpwZM6i8,8125
+official/legacy/detection/ops/nms.py,sha256=Szi_3Z4HshrdqxLhw_NHCvoZ6iqbmq4Nyv7ks48pEpk,8141
 official/legacy/detection/ops/postprocess_ops.py,sha256=1DYKqh7thsJVyORXQqNxKNgfyxxnZAHgAZGO4u7lTjY,21942
 official/legacy/detection/ops/roi_ops.py,sha256=gHwGharQCMsZK-Flam4cZhQM3F7ktxg9ShHzJswM2Ms,22986
 official/legacy/detection/ops/spatial_transform_ops.py,sha256=K1r5Nms7lJlszHkbJHzXru-ba9TiDnIwoEZ6ZV3gXWw,25998
@@ -980,7 +980,7 @@ official/vision/ops/iou_similarity.py,sha256=MHNq3VdcWWVUrybtOfkwX7WTK8B3ZrUQxOT
 official/vision/ops/iou_similarity_test.py,sha256=SMq2wPl98_HenxAKk4p4By_ZoRy6LY7n2CvHCb6QUMg,1898
 official/vision/ops/mask_ops.py,sha256=9yR6KWC9croKvzMtSFUU3cK87-ileUD8MzHfJ-IvBJo,10260
 official/vision/ops/mask_ops_test.py,sha256=34mwpK2-GRS0BWLcOz2Dk2h9kaNBu5zBV70pElpAGsI,2825
-official/vision/ops/nms.py,sha256=HPmgW_fP-NL52mEQXljW1yDPjuHgw1qgh2_avm5jaGw,8099
+official/vision/ops/nms.py,sha256=Q5iGXJ-f_hcLNzyCu7CXKigCK4yHSq40ed7ornqZfVQ,8115
 official/vision/ops/preprocess_ops.py,sha256=HOxLjVVVQejiNENPvrCLlO0so3ds3m1flfrHWvVfM3I,38775
 official/vision/ops/preprocess_ops_3d.py,sha256=aUn1OTLkr1046sAfVefybRxHzhaVAtywlmP0R1UGbIg,15378
 official/vision/ops/preprocess_ops_3d_test.py,sha256=mrs0IVby6WFc2gcD8pIYU__KQFfhmP4m9U000hrpIJQ,7239
@@ -1069,9 +1069,9 @@ tensorflow_models/__init__.py,sha256=021FKgqdPz3ds1xxfV67FWL7e5ECQ7WHbo67D37vAQI
 tensorflow_models/tensorflow_models_test.py,sha256=3oRV5seq-V1La0eY0IFpGLD7AKkiemylW8GyvZIRtmo,1385
 tensorflow_models/nlp/__init__.py,sha256=ro-1L0G8Z1wby8D1Jbaa3No-n73tiNEx7C4f8pAUNlk,807
 tensorflow_models/vision/__init__.py,sha256=3qeLW_6HkgH5hEclFog2DIRu1FSzOr3JynyM23zGhu8,833
-tf_models_nightly-2.11.0.dev20230303.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.11.0.dev20230303.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.11.0.dev20230303.dist-info/METADATA,sha256=yVqNMofIYWntMkNbw0bHF-_2Lq4ODVj8E8n1TqmN-6U,1426
-tf_models_nightly-2.11.0.dev20230303.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.11.0.dev20230303.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.11.0.dev20230303.dist-info/RECORD,,
+tf_models_nightly-2.11.0.dev20230305.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.11.0.dev20230305.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.11.0.dev20230305.dist-info/METADATA,sha256=HEY2SNbGuUN4WXzxUpAoOwQkFBl83MwcYZBBx9eFdb0,1426
+tf_models_nightly-2.11.0.dev20230305.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.11.0.dev20230305.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.11.0.dev20230305.dist-info/RECORD,,