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

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

Резюме

обрезка блоков, которые сильно перекрываются с ранее выбранными блоками. Ограничительные 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 .

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

Operation operation

::tensorflow::Output selected_indices

 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 

 operator::tensorflow::Output() const