参考学习:算法精讲-预测阶段后处理-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单物体预测)
分类:
cv学习 / 处理方法
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· winform 绘制太阳,地球,月球 运作规律
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人