tf.raw_ops.NonMaxSuppressionWithOverlaps

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

tf.raw_ops.NonMaxSuppressionWithOverlaps(
    overlaps, scores, max_output_size, overlap_threshold, score_threshold, name=None
)

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

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

selected_indices = tf.image.non_max_suppression_with_overlaps( overlaps, scores, max_output_size, overlap_threshold, score_threshold) selected_boxes = tf.gather(boxes, selected_indices)