tf.image.non_max_suppression_overlaps
Жадно выбирает подмножество ограничивающих рамок в порядке убывания оценок.
tf.image.non_max_suppression_overlaps(
overlaps,
scores,
max_output_size,
overlap_threshold=0.5,
score_threshold=float('-inf'),
name=None
)
Удаляет рамки, которые сильно пересекаются с ранее выбранными рамками. Значения перекрытия для каждой пары рамок задаются в виде квадратной матрицы n×n. Результатом этой операции является набор целых чисел — индексов выбранных рамок во входной коллекции ограничивающих рамок. Координаты ограничивающих рамок, соответствующие выбранным индексам, можно получить с помощью операции tf.gather. Например:
selected_indices = tf.image.non_max_suppression_overlaps(
overlaps, scores, max_output_size, iou_threshold)
selected_boxes = tf.gather(boxes, selected_indices)
| Аргументы | |
|---|---|
overlaps | Двумерный float Tensor формы [num_boxes, num_boxes], представляющий значения перекрытия рамок для каждой пары в матрице n×n. |
scores | Одномерный float Tensor формы [num_boxes], представляющий оценку для каждой рамки (каждой строки рамок). |
max_output_size | Скалярный целочисленный Tensor, задающий максимальное количество рамок, выбираемых алгоритмом подавления немаксимумов. |
overlap_threshold | Тензор float нулевой размерности, задающий порог для определения чрезмерного перекрытия рамок на основе предоставленных значений перекрытия. |
score_threshold | Тензор float нулевой размерности, задающий порог оценки, ниже которого рамки удаляются. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
selected_indices | Одномерный целочисленный Tensor формы [M], представляющий выбранные индексы из тензора перекрытий, где M <= max_output_size. |
© 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/image/non_max_suppression_overlaps
