参考学习:算法精讲-预测阶段后处理-NMS非极大值抑制_哔哩哔哩_bilibili

以YOLOv1的模型来讲,预测阶段后处理就是把每个bounding box中的每个种类的值算出全概率,再对比bounding box中同种类物品,先设定一个阈值,把bounding box中同种类全概率低于阈值的算为0,再进行一次降序排序,通过遍历,比较第一个与后面所有bounding box预测的IOU(交并比)的值,如果其中存在IOU大于某个阈值,例如设定的0.5,就将全概率小的bounding box清为0,因为IOU大说明,这两个bounding box大概率预测相同的东西在图像上位置重叠,选取预测值大的bounding box进行保留,以此类推,将后续存在值的bounding box一一进行处理,这样图像上划分的bounding box对于一个类别的预测就处理完了,再处理其他类别,处理完所有类别后每个bounding box上除了全部清0的bounding box外,里面最大的值就是这个bbx预测的类别。

(传统的NMS单物体预测)