Attempt to fix gh issue #10330

PiperOrigin-RevId: 405458782

Attempt to fix gh issue #10330
PiperOrigin-RevId: 405458782
12bbefce · Vighnesh Birodkar · TF Object Detection Team · a671a5a6 · 12bbefce
Commit 12bbefce authored Oct 25, 2021 by Vighnesh Birodkar Committed by TF Object Detection Team Oct 25, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/object_detection/utils/ops.py research/object_detection/utils/ops.py +1 -1

No files found.
--- a/research/object_detection/utils/ops.py
+++ b/research/object_detection/utils/ops.py
@@ -811,7 +811,7 @@ def reframe_image_corners_relative_to_boxes(boxes):
  Returns:
    reframed_boxes: Reframes boxes with same shape as input.
  """
-  ymin, xmin, ymax, xmax = tf.unstack(boxes, axis=1)
+  ymin, xmin, ymax, xmax = (boxes[:, 0], boxes[:, 1], boxes[:, 2], boxes[:, 3])

  height = tf.maximum(ymax - ymin, 1e-4)
  width = tf.maximum(xmax - xmin, 1e-4)