tf.image.non_max_suppression_overlaps
| Просмотреть исходный код на GitHub |
Жадно выбирает подмножество прямоугольных областей в порядке убывания оценки.
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 | Матрица с плавающей запятой 2-мерного типа Tensor формы [num_boxes, num_boxes]. |
scores | Массив с плавающей запятой 1-мерного типа Tensor формы [num_boxes] представляющий одиночную оценку, соответствующую каждой прямоугольной области (каждой строке прямоугольных областей). |
max_output_size | Целое число скаляра Tensor представляющее максимальное количество прямоугольных областей, которые необходимо выбрать с помощью подавления не-максимумов. |
overlap_threshold | Число с плавающей запятой, представляющее порог для принятия решения о том, перекрываются ли прямоугольные области слишком сильно по отношению к предоставленным значениям перекрытия. |
score_threshold | Число с плавающей запятой, представляющее порог для принятия решения о том, когда необходимо удалить прямоугольные области на основе оценки. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
selected_indices | Массив целых чисел 1-мерного типа Tensor формы [M] представляющий выбранные индексы из тензора перекрытий, где M <= max_output_size. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/image/non_max_suppression_overlaps