tensorflow::ops::NonMaxSuppression

#include <image_ops.h>

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

Summary

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

Аргументы:

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

Необязательные атрибуты (см. Attrs):

• iou_threshold: Число с плавающей точкой, представляющее порог для определения того, перекрываются ли прямоугольники слишком сильно с точки зрения IOU.

Возвращает:

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

Общедоступные атрибуты

Operation operation

::tensorflow::Output selected_indices

Общедоступные функции

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size
)

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  const NonMaxSuppression::Attrs & attrs
)

::tensorflow::Node * node() const 

operator::tensorflow::Input() const 

operator::tensorflow::Output() const 

Общедоступные статические функции

Attrs IouThreshold(
  float x
)