tensorflow::ops::NonMaxSuppressionV3

#include <image_ops.h>

Жадно выбирает подмножество прямоугольных областей в порядке убывания оценки.

Краткое описание

Удаляет прямоугольные области, имеющие высокое перекрытие (IOU) с ранее выбранными прямоугольными областями. Прямоугольные области с оценкой ниже score_threshold удаляются. Прямоугольные области задаются как [y1, x1, y2, x2], где (y1, x1) и (y2, x2) — координаты любой диагональной пары углов прямоугольной области, и координаты могут быть заданы в нормированном (т. е. лежащем в интервале [0, 1]) или абсолютном виде. Обратите внимание, что этот алгоритм не зависит от расположения начала отсчета в системе координат и, более обще, инвариантен к ортогональным преобразованиям и сдвигам системы координат; таким образом, сдвиг или отражение системы координат приводит к выбору одних и тех же прямоугольных областей алгоритмом. Результатом этой операции является набор целых чисел, индексирующих входной набор прямоугольных областей, представляющих выбранные прямоугольные области. Координаты прямоугольных областей, соответствующие выбранным индексам, можно получить, используя tf.gather operation. Например: selected_indices = tf.image.non_max_suppression_v2( boxes, scores, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather(boxes, selected_indices)

Аргументы:

• scope: Объект Scope
• boxes: 2-мерный тензор с плавающей точкой формы [num_boxes, 4].
• scores: 1-мерный тензор с плавающей точкой формы [num_boxes] представляющий одиночную оценку, соответствующую каждой прямоугольной области (каждая строка boxes).
• max_output_size: скалярный целочисленный тензор, представляющий максимальное количество прямоугольных областей, которые должны быть выбраны с помощью не максимального подавления.
• iou_threshold: 0-мерный тензор с плавающей точкой, представляющий порог для определения того, перекрываются ли прямоугольные области слишком сильно относительно IOU.
• score_threshold: 0-мерный тензор с плавающей точкой, представляющий порог для определения того, когда нужно удалить прямоугольные области на основе оценки.

Возвращаемое значение:

• Output: 1-мерный целочисленный тензор формы [M] представляющий выбранные индексы из тензора boxes, где M <= max_output_size.

Публичные атрибуты

Operation operation

::tensorflow::Output selected_indices

Публичные функции

 NonMaxSuppressionV3(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold
)

::tensorflow::Node * node() const 

operator::tensorflow::Input() const 

operator::tensorflow::Output() const