YOLOv11改进策略【损失函数篇】| 利用MPDIoU，加强边界框回归的准确性

news2024/10/11 3:39:24

一、背景

目标检测和实例分割中的关键问题：
- 现有的大多数边界框回归损失函数在不同的预测结果下可能具有相同的值，这降低了边界框回归的收敛速度和准确性。
现有损失函数的不足：
- 现有的基于 $\ell_n$ 范数的损失函数简单但对各种尺度敏感。
- 当预测框与真实框具有相同的宽高比但不同的宽度和高度值时，现有损失函数可能会存在问题，限制了收敛速度和准确性。

文章目录

一、背景
二、原理
- 2.1 IoU计算原理
- 2.2. 基于最小点距离的IoU度量
- 2.3. 作为边界框回归损失函数
- 2.4 MPDIoU的计算公式
三、添加步骤
- 3.1 ultralytics/utils/metrics.py
- 3.2 修改ultralytics/utils/loss.py
- 3.3 修改ultralytics/utils/tal.py
四、成功运行截图

MPDIoU（Intersection over Union with Minimum Points Distance）是一种用于高效且准确的边界框回归的损失函数。

二、原理

2.1 IoU计算原理

IoU（Intersection over Union）即交并比，用于衡量预测边界框和真实边界框的重合程度。

1. 交集计算：
- 首先确定预测边界框和真实边界框的交集区域。
- 对于两个以左上角和右下角坐标表示的矩形框，分别找出它们在横坐标和纵坐标方向上的重叠区间。
- 如果两个矩形框在横坐标和纵坐标方向上都有重叠部分，那么这个重叠区域就是一个矩形，其面积就是交集的大小。
2. 并集计算：
- 计算预测边界框和真实边界框的并集区域。
- 并集的大小等于两个矩形框各自的面积之和减去它们的交集面积。
3. 比值计算：
- 最后，IoU的值就是交集面积与并集面积的比值。

2.2. 基于最小点距离的IoU度量

原论文中受水平矩形的几何特性启发，设计了一种基于最小点距离的新型IoU度量MPDIoU，直接最小化预测边界框和真实边界框的左上角和右下角点之间的距离。
MPDIoU的计算通过两个任意凸形状 $A$ 和 $B$ ，用其左上角和右下角点的坐标来表示，通过计算两个框的交集与并集之比，再减去左上角和右下角两点距离的归一化值来得到MPDIoU。

2.3. 作为边界框回归损失函数

在训练阶段，通过最小化基于MPDIoU的损失函数 $L_{MPDIoU}=1-MPDIoU$ ，使模型预测的每个边界框 $B_{prd}$ 接近其真实框 $B_{gt}$ 。
现有损失函数中的所有因素（如非重叠区域、中心点距离、宽高偏差等）都可以通过左上角和右下角两点的坐标确定，这意味着提出的 $L_{MPDIoU}$ 不仅考虑了这些因素，还简化了计算过程。

在这里插入图片描述

2.4 MPDIoU的计算公式

MPDIoU的计算公式：
- $MPDIoU=\frac{A\cap B}{A\cup B}-\frac{d_{1}^{2}}{w^{2}+h^{2}}-\frac{d_{2}^{2}}{w^{2}+h^{2}}$
- 其中 $A$ 和 $B$ 是两个任意凸形状， $x_{A1}, y_{A1})$ ， $x_{A2}, y_{A2})$ 表示(A)的左上角和右下角点坐标， $x_{B1}, y_{B1})$ ， $x_{B2}, y_{B2})$ 表示 $B$ 的左上角和右下角点坐标。
- $d_{1}^{2}=(x_{1}^{B}-x_{1}^{A})^{2}+(y_{1}^{B}-y_{1}^{A})^{2}$ ， $d_{2}^{2}=(x_{2}^{B}-x_{2}^{A})^{2}+(y_{2}^{B}-y_{2}^{A})^{2}$ 。
基于MPDIoU的损失函数计算公式：
- $L_{MPDIoU}=1-MPDIoU$ 。

三、添加步骤

3.1 ultralytics/utils/metrics.py

此处需要查看的文件是ultralytics/utils/metrics.py

metrics.py中定义了模型的损失函数和计算方法，我们想要加入新的损失函数就只需要将代码放到这个文件内即可

将下方的bbox_iou函数替换原本的bbox_iou函数即可：

def bbox_iou(box1, box2, xywh=True, GIoU=False, DIoU=False, CIoU=False, MDPIoU=False, feat_h=640, feat_w=640, eps=1e-7):
    # Returns Intersection over Union (IoU) of box1(1,4) to box2(n,4)

    # Get the coordinates of bounding boxes
    if xywh:  # transform from xywh to xyxy
        (x1, y1, w1, h1), (x2, y2, w2, h2) = box1.chunk(4, -1), box2.chunk(4, -1)
        w1_, h1_, w2_, h2_ = w1 / 2, h1 / 2, w2 / 2, h2 / 2
        b1_x1, b1_x2, b1_y1, b1_y2 = x1 - w1_, x1 + w1_, y1 - h1_, y1 + h1_
        b2_x1, b2_x2, b2_y1, b2_y2 = x2 - w2_, x2 + w2_, y2 - h2_, y2 + h2_
    else:  # x1, y1, x2, y2 = box1
        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 + eps
        w2, h2 = b2_x2 - b2_x1, b2_y2 - b2_y1 + eps

    # Intersection area
    inter = (torch.min(b1_x2, b2_x2) - torch.max(b1_x1, b2_x1)).clamp(0) * \
            (torch.min(b1_y2, b2_y2) - torch.max(b1_y1, b2_y1)).clamp(0)

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

    # IoU
    iou = inter / union
    if CIoU or DIoU or GIoU:
        cw = torch.max(b1_x2, b2_x2) - torch.min(b1_x1, b2_x1)  # convex (smallest enclosing box) width
        ch = torch.max(b1_y2, b2_y2) - torch.min(b1_y1, b2_y1)  # convex height
        if CIoU or DIoU:  # Distance or Complete IoU https://arxiv.org/abs/1911.08287v1
            c2 = cw ** 2 + ch ** 2 + eps  # convex diagonal squared
            rho2 = ((b2_x1 + b2_x2 - b1_x1 - b1_x2) ** 2 + (b2_y1 + b2_y2 - b1_y1 - b1_y2) ** 2) / 4  # center dist ** 2
            if CIoU:  # https://github.com/Zzh-tju/DIoU-SSD-pytorch/blob/master/utils/box/box_utils.py#L47
                v = (4 / math.pi ** 2) * torch.pow(torch.atan(w2 / h2) - torch.atan(w1 / h1), 2)
                with torch.no_grad():
                    alpha = v / (v - iou + (1 + eps))
                return iou - (rho2 / c2 + v * alpha)  # CIoU
            return iou - rho2 / c2  # DIoU
        c_area = cw * ch + eps  # convex area
        return iou - (c_area - union) / c_area  # GIoU https://arxiv.org/pdf/1902.09630.pdf
    elif MDPIoU:
        d1 = (b2_x1 - b1_x1) ** 2 + (b2_y1 - b1_y1) ** 2
        d2 = (b2_x2 - b1_x2) ** 2 + (b2_y2 - b1_y2) ** 2
        mpdiou_hw_pow = feat_h ** 2 + feat_w ** 2
        return iou - d1 / mpdiou_hw_pow - d2 / mpdiou_hw_pow  # MPDIoU
    return iou  # IoU

在这里插入图片描述

3.2 修改ultralytics/utils/loss.py

utils\loss.py用于计算各种损失。

在BboxLoss函数内修改如下代码，使模型调用此MPDIoU损失函数。


iou = bbox_iou(pred_bboxes[fg_mask], target_bboxes[fg_mask], xywh=False, MDPIoU=True)

在这里插入图片描述

3.3 修改ultralytics/utils/tal.py

tal.py中是一些损失函数的功能应用。

在iou_calculation函数内修改如下代码，使模型调用此MPDIoU损失函数。

return bbox_iou(gt_bboxes, pd_bboxes, xywh=False, MPDIoU=True).squeeze(-1).clamp_(0)

在这里插入图片描述

四、成功运行截图

在这里插入图片描述

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/2203959.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！