From 60da17a2bcb769ddf02cf8373d2cd19cd1db1e2b Mon Sep 17 00:00:00 2001 From: wbenbihi Date: Sat, 20 Aug 2022 17:09:51 +0800 Subject: [PATCH] [FIX] bad argument name in tf_reshape --- hourglass_tensorflow/utils/tf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hourglass_tensorflow/utils/tf.py b/hourglass_tensorflow/utils/tf.py index 5a8af9e..f83957d 100644 --- a/hourglass_tensorflow/utils/tf.py +++ b/hourglass_tensorflow/utils/tf.py @@ -165,7 +165,7 @@ def tf_expand_bbox( bottom_right_y + height * (bbox_factor - 1.0), ) return tf.cast( - tf_reshape_slice([new_tl_x, new_tl_y, new_br_x, new_br_y], dim=2, **kwargs), + tf_reshape_slice([new_tl_x, new_tl_y, new_br_x, new_br_y], shape=2, **kwargs), dtype=tf.int32, )