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 и ряда ограничений.

Эта операция возвращает одну ограничивающую рамку, которую можно использовать для обрезки исходного изображения. Результат возвращается в виде трех тензоров: begin, size и bboxes. Первые два тензора можно напрямую передать в 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 и ограничивающие рамки не переданы, возникает ошибка.