Skip to content

GitLab

Commit 0e0f739b authored by syiming's avatar syiming
Browse files

force pad to multiple pad tensors of zero with same dtype of input

parent 26f976c1
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/object_detection/utils/ops.py
View file @ 0e0f739b
...@@ -216,13 +216,13 @@ def pad_to_multiple(tensor, multiple): ...@@ -216,13 +216,13 @@ def pad_to_multiple(tensor, multiple):
height_pad = tf.zeros([ height_pad = tf.zeros([
batch_size, padded_tensor_height - tensor_height, tensor_width, batch_size, padded_tensor_height - tensor_height, tensor_width,
tensor_depth tensor_depth
]) ], dtype=tensor.dtype)
tensor = tf.concat([tensor, height_pad], 1) tensor = tf.concat([tensor, height_pad], 1)
if padded_tensor_width != tensor_width: if padded_tensor_width != tensor_width:
width_pad = tf.zeros([ width_pad = tf.zeros([
batch_size, padded_tensor_height, padded_tensor_width - tensor_width, batch_size, padded_tensor_height, padded_tensor_width - tensor_width,
tensor_depth tensor_depth
]) ], dtype=tensor.dtype)
tensor = tf.concat([tensor, width_pad], 2) tensor = tf.concat([tensor, width_pad], 2)
return tensor return tensor
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment