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 и рамки не предоставлены, генерируется ошибка.