YOLOV5改进（Soft-NMS（多种IOU））

最新推荐文章于 2025-05-08 09:09:11 发布

eagleflying_cau

最新推荐文章于 2025-05-08 09:09:11 发布

阅读量1.4k

点赞数 4

CC 4.0 BY-SA版权

文章标签： YOLO

本文链接：https://blog.csdn.net/eagleflying_cau/article/details/131146782

文章介绍了一种在密集数据集上用于目标检测的非极大值抑制(NMS)方法，通过引入不同的IoU变体如CIOU、DIOU、SIOU、GIOU和EIOU来改进检测性能。在训练时使用这些改进的IoU计算，而在验证时切换回基本的IoU。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.适合在密集的数据集上进行检测

2.改进方法

（1）将以下代码复制到utils/general.py中

def box_iou_for_nms(box1, box2, GIoU=False, DIoU=False, CIoU=False, SIoU=False, EIou=False, eps=1e-7):
    # Returns Intersection over Union (IoU) of box1(1,4) to box2(n,4)

    b1_x1, b1_y1, b1_x2, b1_y2 = box1.chunk(4, -1)
    b2_x1, b2_y1, b2_x2, b2_y2 = box2.chunk(4, -1)
    w1, h1 = b1_x2 - b1_x1, (b1_y2 - b1_y1).clamp(eps)
    w2, h2 = b2_x2 - b2_x1, (b2_y2 - b2_y1).clamp(eps)

    # Intersection area
    inter = (b1_x2.minimum(b2_x2) - b1_x1.maximum(b2_x1)).clamp(0) * \
            (b1_y2.minimum(b2_y2) - b1_y1.maximum(b2_y1)).clamp(0)

    # Union Area
    union = w1 * h1 + w2 * h2 - inter + eps

    # IoU
    iou = inter / union
    if CIoU or DIoU or GIoU or EIou:
        cw = b1_x2.maximum(b2_x2) - b1_x1.minimum(b2_x1)  # convex (smallest enclosing box) width
        ch = b1_y2.maximum(b2_