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

Например, если изображение имеет размер 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]` содержащий прямоугольники.
`name`	Имя операции (необязательно).
`colors`	A `Tensor` типа `float32`. 2-мерный. Список RGBA цветов для циклической закраски прямоугольников.

Возвращаемое значение
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/compat/v1/image/draw_bounding_boxes