NMS算法简介 NMS(非极大值抑制)是目标检测中的后处理算法,用来过滤冗余的框。 算法步骤如下: 对候选框按照置信度进行排序 选择置信度最高的框放进结果框,并与其他框计算iou,去除iou大于阈值的框 重复2,直至候选框为空 NMS算法python代码实现 def nms(boxes, iou_thres): """ 非极大值抑制 """ x1 = boxes[:, 0] y1 = boxes[:, 1] x2 = boxes[:, 2] y2 = boxes[:, 3