tf.image.pad_to_bounding_box
| Просмотреть исходный код на GitHub |
Заполняет image нулями до указанных height и width.
tf.image.pad_to_bounding_box(
image, offset_height, offset_width, target_height, target_width
)
Добавляет offset_height строк нулей сверху, offset_width столбцов нулей слева, а затем заполняет изображение нулями снизу и справа до тех пор, пока оно не будет иметь размер target_height, target_width.
Этот оператор ничего не делает, если offset_* равно нулю, и изображение уже имеет размер target_height на target_width.
Пример использования:
x = [[[1., 2., 3.],
[4., 5., 6.]],
[[7., 8., 9.],
[10., 11., 12.]]]
padded_image = tf.image.pad_to_bounding_box(x, 1, 1, 4, 4)
padded_image
<tf.Tensor: shape=(4, 4, 3), dtype=float32, numpy=
array([[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 1., 2., 3.],
[ 4., 5., 6.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 7., 8., 9.],
[10., 11., 12.],
[ 0., 0., 0.]],
[[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.],
[ 0., 0., 0.]]], dtype=float32)>
| Аргументы | |
|---|---|
image | 4-мерный тензор формы [batch, height, width, channels] или 3-мерный тензор формы [height, width, channels]. |
offset_height | Количество строк нулей, добавляемых сверху. |
offset_width | Количество столбцов нулей, добавляемых слева. |
target_height | Высота выходного изображения. |
target_width | Ширина выходного изображения. |
| Возвращаемое значение | |
|---|---|
Если image был 4-мерным, 4-мерный float тензор формы [batch, target_height, target_width, channels] Если image был 3-мерным, 3-мерный float тензор формы [target_height, target_width, channels] |
| Исключения | |
|---|---|
ValueError | Если форма image несовместима с аргументами offset_* или target_*, или любой из аргументов offset_height или offset_width отрицательный. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/image/pad_to_bounding_box