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

#include <image_ops.h>

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

Резюме

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

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

selected_indices = tf.image.non_max_suppression_with_overlaps (перекрытия, оценки, max_output_size, overlap_threshold, score_threshold) selected_boxes = tf.gather (блоки, selected_indices)

Аргументы:

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

Возврат:

  • Output : 1-мерный целочисленный тензор формы [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

selected_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 

оператор :: тензорный поток :: Ввод

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const