非极大抑制算法 (Non-Maximum Suppression),是一种常用的对于bounding box 进行筛选的方法。如图所示:对于图中的同一物体(猫)产生了多个bounding box,我们如何选择一个最适合的呢?这时候就需要应用NMS算法了。
一般来说bounding box会有一些属性值:bbox 的中心位置(x,y),长宽(h,w),bbox中是否有物体(confidence score[0,1],越大越确定为物体),bbox对应各个类别(猫狗)的概率。
算法流程
看了是不是很懵?让我们来具体操作一下:
:5个颜色所代表的bbox
:每个bbox对应的confidence score
: Threshold of IoU(0.5)
IoU是指:
操作流程如下:
如果我们把 Threshold 调高一点:
: Threshold of IoU(0.7)
由上可知Threshold of IoU设置太高可能会造成物体的重复检查。
小伙伴们如果觉得文章还行的请点个赞呦!!同时觉得文章哪里有问题的可以评论一下 谢谢你!