Non-maxing supressing非最大值抑制

来源:互联网 发布:深圳云计算招聘岗位 编辑:程序博客网 时间:2024/06/07 01:24

SSD物体检测中选出一堆的候选框,需要最大值抑制消除一些相似的。ssd网络是会预测一些候选框,并给出所得到候选框的置信度confidence,然后对得到的框的中心和边长进行微调的网络。
NMS具体的做法选择一个confidence最大的框,然后比较相似度,将重叠度高IOU的框删除。

不过在做的时候,踩了几个坑,围观了,又有坑可踩。

1 图片中相同的物体可能有好几个,所以直接选最大的,将所有的和他比较,肯定会没办法抑制其他相同物体的框。
方法就是如果出现这中情况,就需要再选择剩下中概率最大的一个框。

这是我写的nms的github链接,里面还有用到PIL库显示图片的代码

原创粉丝点击