tf.compat.v1.image.draw_bounding_boxes

Рисует ограничивающие рамки на пакете изображений.

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

Возвращает
Объект типа `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/compat/v1/image/draw_bounding_boxes