tf.raw_ops.NonMaxSuppression

Жадно выбирает подмножество прямоугольников по убыванию оценки,

Просмотр псевдонимов

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.raw_ops.NonMaxSuppression

tf.raw_ops.NonMaxSuppression(
    boxes, scores, max_output_size, iou_threshold=0.5, name=None
)

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

Аргументы
`boxes`	A `Tensor` типа `float32`. Двухмерный тензор с плавающей точкой формы `[num_boxes, 4]`.
`scores`	A `Tensor` типа `float32`. Одномерный тензор с плавающей точкой формы `[num_boxes]`, представляющий одиночную оценку, соответствующую каждому прямоугольнику (каждой строке прямоугольников).
`max_output_size`	A `Tensor` типа `int32`. Скалярный целочисленный тензор, представляющий максимальное количество прямоугольников, которые должны быть выбраны с помощью подавления не максимальных значений.
`iou_threshold`	Необязательный `float`. По умолчанию `0.5`. Число с плавающей точкой, представляющее порог для определения того, перекрываются ли прямоугольники слишком сильно с точки зрения IOU.
`name`	Имя операции (необязательно).

Возвращаемое значение
A `Tensor` типа `int32`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/NonMaxSuppression