diff options
Diffstat (limited to 'Code/tfutils.py')
| -rw-r--r-- | Code/tfutils.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/Code/tfutils.py b/Code/tfutils.py index 22baf95..0d2d8dd 100644 --- a/Code/tfutils.py +++ b/Code/tfutils.py @@ -92,12 +92,12 @@ def batch_pad_to_bounding_box(images, offset_height, offset_width, target_height rpad = np.zeros([batch_size, target_height, num_rpad, channels]) padded = images - if num_tpad > 0 and num_bpad > 0: padded = tf.concat(1, [tpad, padded, bpad]) - elif num_tpad > 0: padded = tf.concat(1, [tpad, padded]) - elif num_bpad > 0: padded = tf.concat(1, [padded, bpad]) - if num_lpad > 0 and num_rpad > 0: padded = tf.concat(2, [lpad, padded, rpad]) - elif num_lpad > 0: padded = tf.concat(2, [lpad, padded]) - elif num_rpad > 0: padded = tf.concat(2, [padded, rpad]) + if num_tpad > 0 and num_bpad > 0: padded = tf.concat(axis=1, values=[tpad, padded, bpad]) + elif num_tpad > 0: padded = tf.concat(axis=1, values=[tpad, padded]) + elif num_bpad > 0: padded = tf.concat(axis=1, values=[padded, bpad]) + if num_lpad > 0 and num_rpad > 0: padded = tf.concat(axis=2, values=[lpad, padded, rpad]) + elif num_lpad > 0: padded = tf.concat(axis=2, values=[lpad, padded]) + elif num_rpad > 0: padded = tf.concat(axis=2, values=[padded, rpad]) return padded |