yolov5多个框重叠问题

2023-10-23 19:29:57 浏览数 (3)

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

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

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

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

0 人点赞