tf.image.draw_bounding_boxes

Рисует рамки обнаружения на наборе изображений.

tf.image.draw_bounding_boxes(
    images, boxes, colors, name=None
)

Возвращает копию images, но накладывает на неё одну или несколько рамок обнаружения, заданных координатами в boxes. Координаты каждой рамки обнаружения в boxes закодированы как [y_min, x_min, y_max, x_max]. Координаты рамок обнаружения — это числа с плавающей точкой в [0.0, 1.0], относительные к ширине и высоте исходного изображения.

Например, если изображение имеет размер 100 x 200 пикселей (высота x ширина), и рамка обнаружения задана как [0.1, 0.2, 0.5, 0.9], то координаты верхнего левого и нижнего правого углов рамки обнаружения будут (40, 10) и (180, 50) (в координатах (x, y)).

Часть рамки обнаружения может выходить за пределы изображения.

Аргументы
`images`	A `Tensor`. Должно быть одного из следующих типов: `float32`, `half`. 4-мерное с формой `[batch, height, width, depth]`. Набор изображений.
`boxes`	A `Tensor` типа `float32`. 3-мерное с формой `[batch, num_bounding_boxes, 4]` содержащее рамки обнаружения.
`colors`	A `Tensor` типа `float32`. 2-мерное. Список RGBA цветов для циклического выбора при рисовании рамок.
`name`	Имя операции (необязательно).

Возвращаемое значение
A `Tensor`. Имеет тот же тип, что и `images`.

Пример использования:

# create an empty image
img = tf.zeros([1, 3, 3, 3])
# draw a box around the image
box = np.array([0, 0, 1, 1])
boxes = box.reshape([1, 1, 4])
# alternate between red and blue
colors = np.array([[1.0, 0.0, 0.0], [0.0, 0.0, 1.0]])
tf.image.draw_bounding_boxes(img, boxes, colors)
<tf.Tensor: shape=(1, 3, 3, 3), dtype=float32, numpy=
array([[[[1., 0., 0.],
        [1., 0., 0.],
        [1., 0., 0.]],
        [[1., 0., 0.],
        [0., 0., 0.],
        [1., 0., 0.]],
        [[1., 0., 0.],
        [1., 0., 0.],
        [1., 0., 0.]]]], dtype=float32)>

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/image/draw_bounding_boxes