тензорный поток :: ops :: КомбинированныйNonMaxSuppression

#include <image_ops.h>

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

Резюме

Эта операция выполняет non_max_suppression для входных данных в пакете для всех классов. Удаляет блоки, которые имеют высокое перекрытие с пересечением по объединению (IOU) с ранее выбранными блоками. Ограничивающие прямоугольники представлены как [y1, x1, y2, x2], где (y1, x1) и (y2, x2) - координаты любой диагональной пары углов прямоугольника, и координаты могут быть предоставлены как нормализованные (т.е. лежащие в интервал [0, 1]) или абсолютный. Обратите внимание, что этот алгоритм не зависит от того, где начало координат находится в системе координат. Также обратите внимание, что этот алгоритм инвариантен к ортогональным преобразованиям и переносам системы координат; таким образом, перевод или отражение системы координат приводят к тому, что алгоритм выбирает одни и те же блоки. Результатом этой операции является окончательный тензор полей, оценок и классов, возвращаемый после выполнения non_max_suppression.

Аргументы:

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

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

  • pad_per_class: Если false, выходные поля, оценки и классы max_total_size дополняются / max_total_size до max_total_size . Если true, выходные поля, оценки и классы nmsed дополняются, чтобы иметь длину max_size_per_class * num_classes , если она не превышает max_total_size в этом случае она обрезается до max_total_size . По умолчанию false.
  • clip_boxes: если true, предположим, что координаты блока находятся между [0, 1] и обрезаем блоки вывода, если они выходят за пределы [0, 1]. Если false, не обрезать и выводить координаты блока как есть.

Возврат:

  • Output nmsed_boxes: тензор [batch_size, max_detections, 4] float32, содержащий не-max подавленные боксы.
  • Output nmsed_scores: тензор [batch_size, max_detections] float32, содержащий оценки для боксов.
  • Output nmsed_classes: тензор [batch_size, max_detections] float32, содержащий классы для боксов.
  • Output valid_detections: тензор [batch_size] int32, указывающий количество допустимых обнаружений на элемент пакета. Действительны только верхние записи num_detections [i] в ​​nms_boxes [i], nms_scores [i] и nms_class [i]. Остальные записи - нулевые отступы.

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

CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)
CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, const CombinedNonMaxSuppression::Attrs & attrs)

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

nmsed_boxes
nmsed_classes
nmsed_scores
operation
valid_detections

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

ClipBoxes (bool x)
PadPerClass (bool x)

Структуры

tenorflow :: ops :: CombinedNonMaxSuppression :: Attrs

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

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

nmsed_boxes

::tensorflow::Output nmsed_boxes

nmsed_classes

::tensorflow::Output nmsed_classes

nmsed_scores

::tensorflow::Output nmsed_scores

операция

Operation operation

valid_detections

::tensorflow::Output valid_detections

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

КомбинированныйNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold
)

CombinedNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const CombinedNonMaxSuppression::Attrs & attrs
)

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

ClipBoxes

Attrs ClipBoxes(
  bool x
)

PadPerClass

Attrs PadPerClass(
  bool x
)