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] относительно ширины и высоты исходного изображения.

Например, если изображение имеет размер 100x200 пикселей (высота 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)>

© 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/draw_bounding_boxes