tf.image.sample_distorted_bounding_box

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

tf.image.sample_distorted_bounding_box(
    image_size, bounding_boxes, seed=0, min_object_covered=0.1,
    aspect_ratio_range=None, area_range=None, max_attempts=None,
    use_image_if_no_bounding_boxes=None, 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,
    min_object_covered=0.1)

# 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.compat.v1.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 и рамки обводки не указаны, возникает ошибка.