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

Возвращаемое значение
`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/api_docs/python/tf/image/draw_bounding_boxes