weixin_47744078的博客

原创 YOLOv1中的NMS（非极大值抑制）详解

对于其中的每一个类别，首先会进行排序操作，按照置信度（全概率得分）从大到小进行排序，将该类别下置信度最大的bounding box依次和比它小的的bounding box进行比较，计算他们之间的iou（交并比，一定程度上反映两个边界框之间的重叠程度），此时也会给定一个iou的阀值，当两个边界框的iou值超过给定的阀值，则说明第二个大的边界框可能是重复识别了同一个物体，则将其抹去删除，保留置信度（全概率得分）最大的那个。对每个类别都依次重复上述的操作，最终得到一个包含多个0的稀疏矩阵，如下图所示。