yolov5多个框重叠问题

NMS(Non-Maximum Suppression,非极大值抑制)是一种在计算机视觉和目标检测领域常用的技术。它通常用于在图像或视频中找出物体或目标的位置,并剔除重叠的边界框,以确保最终的检测结果准确且不重叠。 

会出现多个框重叠的问题其实就是nms非极大值抑制iou阈值的问题,可以把参数iou改变一点试试,我直接改成0的效果

可见重叠框大大减少了,为什么会这样子呢?这是因为Iou是两个框边界的交集和并集之比,那么这个iou越大说明两个框越接近。Yolo中使用的nms非极大值抑制算法,就是用来解决一个目标被多次检测的问题的,nms首先从所有的框中选出置信度最大的框,然后计算剩余的检测框与该框的iou,如果iou大于所设置的iou_thres,即这两个框重叠的越多,就会去掉这个框。

所以如果想要去掉重叠的框,我们可以把参数iou_thres改小一点,从而去掉更多重叠的框。 

如果是train时产生的多个框重叠问题,请修改train.py的这个地方

如果是val或者detec时出现的问题,请修改val.py或detect.py的iou_thres参数

  • 6
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
YOLOv5中的多类别NMS(非极大值抑制)是为了让每个类别都能独立执行NMS,并确保不同类别的边不会重叠。实现方法是在所有边上添加一个偏移量,该偏移量仅根据类别的索引值来确定,并且足够大,以确保来自不同类别的边不会发生重叠。这种处理方式可以通过调用PyTorch的torchvision.ops.batched_nms实现。 在YOLOv5中,可以通过终端命令传递参数来确定是执行多个类一起应用NMS还是按照不同的类别分别应用NMS。具体地,在执行命令行时,可以使用参数"--agnostic-nms"来决定是否执行多个类一起应用NMS。如果传入了"--agnostic-nms"参数,那么就会执行多个类一起应用NMS;反之,则会按照不同的类别分别应用NMS。 在具体实现中,使用了带有偏移量的边进行NMS。这些偏移量是根据类别的ID确定的,用于确保来自不同类别的边不会重叠。同时,还需要传入边的坐标和分数,并设置一个IoU阈值来进行非极大值抑制。这里的nms调用了torchvision下的nms函数来执行NMS操作。 总结起来,YOLOv5中的多类别NMS是为了让每个类别都能独立执行NMS,并避免不同类别的边重叠。通过在边上添加偏移量,并根据类别的ID确定偏移量的大小,可以实现多类别NMS的处理。可以通过传递参数来确定是执行多个类一起应用NMS还是按照不同的类别分别应用NMS。<span class="em">1</span><span class="em">2</span><span class="em">3</span><span class="em">4</span>

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

MaolinYe(叶茂林)

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值