тензорный поток :: ops :: NonMaxSuppression

#include <image_ops.h>

Жадно выбирает подмножество ограничивающих рамок в порядке убывания оценки.

Резюме

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

Аргументы:

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

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

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

Возврат:

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

Конструкторы и деструкторы

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)

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

operation
selected_indices

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

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

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

IouThreshold (float x)

Структуры

tenorflow :: ops :: NonMaxSuppression :: Attrs

Необязательные установщики атрибутов для NonMaxSuppression .

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

операция

Operation operation

selected_indices

::tensorflow::Output selected_indices

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

NonMaxSuppression

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

NonMaxSuppression

 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 

оператор :: tenorflow :: Выход

 operator::tensorflow::Output() const 

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

IouThreshold

Attrs IouThreshold(
  float x
)