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
)

Аннотации границ часто предоставляются в дополнение к истинным меткам в задачах распознавания изображений или локализации объектов. Общим методом обучения такой системы является случайное искажение изображения при сохранении его содержимого, т.е. усиление данных. Данный оператор выводит случайное искажённое расположение объекта, т.е. рамку, заданную 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 и рамки не заданы, возникает ошибка.