tf.raw_ops.SampleDistortedBoundingBox

Генерация одного случайно искаженного ограничивающего прямоугольника для изображения.

tf.raw_ops.SampleDistortedBoundingBox(
    image_size,
    bounding_boxes,
    seed=0,
    seed2=0,
    min_object_covered=0.1,
    aspect_ratio_range=[0.75, 1.33],
    area_range=[0.05, 1],
    max_attempts=100,
    use_image_if_no_bounding_boxes=False,
    name=None
)

Аннотации ограничивающих прямоугольников часто предоставляются дополнительно к меткам ground-truth в задачах распознавания изображений или локализации объектов. Распространенным методом обучения такой системы является случайное искажение изображения, сохраняя при этом его содержание, т. е. усиление данных. Этот оператор выводит случайное искаженное расположение объекта, т. е. ограничивающий прямоугольник, учитывая image_size, bounding_boxes и ряд ограничений.

Выходной результат этого оператора — один ограничивающий прямоугольник, который можно использовать для обрезки исходного изображения. Выход возвращается как 3 тензора: begin, size и bboxes. Первые 2 тензора можно напрямую передать в tf.slice для обрезки изображения. Последний можно передать в tf.image.draw_bounding_boxes для визуализации ограничивающего прямоугольника.

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

Например,

# Generate a single distorted bounding box.
begin, size, bbox_for_draw = tf.image.sample_distorted_bounding_box(
    tf.shape(image),
    bounding_boxes=bounding_boxes)

# Draw the bounding box in an image summary.
image_with_box = tf.image.draw_bounding_boxes(tf.expand_dims(image, 0),
                                              bbox_for_draw)
tf.summary.image('images_with_box', image_with_box)

# Employ the bounding box to distort the image.
distorted_image = tf.slice(image, begin, size)

Обратите внимание, что если информация об ограничивающем прямоугольнике недоступна, установка use_image_if_no_bounding_boxes = true предполагает наличие одного неявного ограничивающего прямоугольника, охватывающего всё изображение. Если use_image_if_no_bounding_boxes имеет значение false и ограничивающие прямоугольники не предоставлены, генерируется ошибка.