NMS函数python

NMS函数是指非极大值抑制函数，用于在目标检测中去除重叠的边界框。下面是一个简洁而靠近底层的Python实现示例：

def nms(boxes, scores, threshold):
    sorted_indices = sorted(range(len(scores)), key=lambda i: scores[i], reverse=True)
    selected_indices = []
    while len(sorted_indices) > 0:
        best_index = sorted_indices[0]
        selected_indices.append(best_index)
        remaining_indices = []
        for i in range(1, len(sorted_indices)):
            index = sorted_indices[i]
            if iou(boxes[best_index], boxes[index]) < threshold:
                remaining_indices.append(index)
        sorted_indices = remaining_indices
    return selected_indices

其中，`boxes`是边界框的坐标，`scores`是对应的置信度，`threshold`是IOU阈值，`iou`是计算两个边界框之间的交并比的函数。

该函数的实现原理是根据置信度对边界框进行排序，选取置信度最高的边界框作为基准，然后依次比较其他边界框与基准边界框的重叠度(IOU)，如果重叠度超过阈值，则将其舍去。重复这个过程直到没有边界框剩余。