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

#include <image_ops.h>

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

Резюме

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

Аргументы:

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

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

  • pad_to_max_output_size: если true, выходные selected_indices дополняются до длины max_output_size . По умолчанию false.

Возврат:

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

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

NonMaxSuppressionV4 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)
NonMaxSuppressionV4 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, const NonMaxSuppressionV4::Attrs & attrs)

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

operation
selected_indices
valid_outputs

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

PadToMaxOutputSize (bool x)

Структуры

tenorflow :: ops :: NonMaxSuppressionV4 :: Attrs

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

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

операция

Operation operation

selected_indices

::tensorflow::Output selected_indices

valid_outputs

::tensorflow::Output valid_outputs

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

NonMaxSuppressionV4

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

NonMaxSuppressionV4

 NonMaxSuppressionV4(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const NonMaxSuppressionV4::Attrs & attrs
)

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

PadToMaxOutputSize

Attrs PadToMaxOutputSize(
  bool x
)