深度学习代码分享
pontoon
这个作者很懒,什么都没留下…
展开
-
soft-nms numpy-python实现软非极大值抑制精简代码
soft-nms相对于nms整个改进只需要使用绿色虚线表示的Soft-NMS替换红色虚线表示的NMS。B集合是检测到的所有建议框,S集合是各个建议框得分(分数是指建议框包含物体的可能性大小),Nt是指手动设置的阈值。M为当前得分最高框,bi 为待处理框。 相比其他网站的代码,此soft-nms代码实现非常精短,先给出iou的代码: def iou(bbox,gt): #lt是两个框中间重叠框的最左边和最上边的坐标,rb是两个框中间重叠框的最右边和最下边的坐标 lt = np..原创 2022-03-08 11:09:34 · 4375 阅读 · 3 评论 -
NMS-python-numpy 精简非极大值抑制代码
参考别人的代码,发现别人的代码比较繁琐,以下是精简后的NMS代码,代码可读性强,最后给出可视化图以及可视化代码(可视化是参考别人的代码) NMS过程: 1 将各组box按照score降序排列 2 从score最大值开始,置为当前box,保存idex,然后依次遍历后面的box,计算与当前box的IOU值,若大于阈值,则抑制,不会输出 3 完成一轮遍历后,继续选择下一个非抑制的box作为当前box,重复步骤2 4 返回没有被抑制的index即符合条件的box NMS在过程中需要计算iou,所以原创 2022-03-08 10:01:56 · 636 阅读 · 0 评论