тензорный поток:: опс:: NonMaxSuppressionWithOverlaps

#include <image_ops.h>

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

Краткое содержание

обрезка полей, которые сильно перекрываются с ранее выбранными полями. Ограничительные рамки со счетом меньше, чем score_threshold удаляются. Значения перекрытия Nxn предоставляются в виде квадратной матрицы, что позволяет определить собственный критерий перекрытия (например, пересечение по объединению, пересечение по площади и т. д.).

Результатом этой операции является набор целых чисел, индексируемых во входную коллекцию ограничивающих рамок, представляющих выбранные рамки. Координаты ограничивающего прямоугольника, соответствующие выбранным индексам, затем можно получить с помощью tf.gather operation . Например:

selected_indices = tf.image.non_max_suppression_with_overlaps(перекрытия, оценки, максимальный_выходной_размер, перекрытие_порог, оценка_порог) selected_boxes = tf.gather(boxes, selected_indices)

Аргументы:

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

Возврат:

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

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

NonMaxSuppressionWithOverlaps (const :: tensorflow::Scope & scope, :: tensorflow::Input overlaps, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input overlap_threshold, :: tensorflow::Input score_threshold)

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

operation
selected_indices

Общественные функции

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

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

операция

Operation operation

выбранные_indices

::tensorflow::Output selected_indices

Общественные функции

NonMaxSuppressionWithOverlaps

 NonMaxSuppressionWithOverlaps(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input overlaps,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input overlap_threshold,
  ::tensorflow::Input score_threshold
)

узел

::tensorflow::Node * node() const 

оператор::tensorflow::Input

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const