tf.raw_ops.SampleDistortedBoundingBoxV2

Генерация одной случайной искажённой рамки для изображения.

tf.raw_ops.SampleDistortedBoundingBoxV2(
    image_size, bounding_boxes, min_object_covered, seed=0, seed2=0,
    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 и рамки не предоставлены, возникает ошибка.