改进YOLOv7的wiou损失函数：融合最新的Wise-IoU损失函数，提升性能｜超越CIoU、SIoU，助力YOLOv7模型提升准确率

### Wise-IoU 损失函数的原理 Wise-IoU (Weighted Intersection over Union, WIoU) 是一种改进型的目标检测损失函数，旨在解决传统 IoU 在处理不同大小目标时存在的局限性。它通过对重叠区域赋予不同的权重来增强模型对于小目标和大目标的适应能力。 #### 1. **基本定义** 传统的 IoU 定义为目标预测框与真实框之间的交集面积除以并集面积。然而，这种简单的度量方式可能无法很好地反映边界框回归的真实误差情况，尤其是在目标尺寸差异较大时。因此，WIoU 引入了一种加权机制，使得损失更加关注于那些更难拟合的部分[^1]。 其核心公式可以表示为： \[ L_{WIoU} = \frac{\sum w_i(A_i \cap B_i)}{\sum w_i(A_i \cup B_i)} \] 其中 \(w_i\) 表示第 i 对预测框和真实框之间的一个动态权重因子，\(A_i\) 和 \(B_i\) 则分别代表预测框和真实框。 --- #### 2. **权重的设计原则** 为了更好地适配不同尺度的目标，WIoU 中的权重通常基于以下几个因素进行调整： - **目标尺寸**：较大的目标往往更容易被准确预测，而较小的目标则更具挑战性。因此，可以通过设置更大的权重给小目标来平衡这一现象。 \[ w_i = f(\text{size}(A_i), \text{size}(B_i)) \] 这里 \(f\) 是一个单调递减函数，随着目标尺寸增加而逐渐降低权重值[^3]。 - **位置偏差**：当两个框的位置偏离较远时，应该给予更高的惩罚力度；反之亦然。这有助于引导网络更快地收敛到正确的解空间。 --- #### 3. **与其他变体的区别** | 名称 | 特点 | |------------|----------------------------------------------------------------------------------------| | IoU | 单纯衡量交并比 | | GIoU | 考虑外部最小闭包矩形的影响 | | DIoU | 加入中心点距离作为额外约束条件 | | CIoU/EIoU | 进一步细化考虑宽高比例关系 | | Wise-IoU | 动态分配权重，特别强调对困难样本（如小物体）的关注 | 相比其他变体，WIoU 不仅继承了它们的优点，还在实际应用中表现出更强鲁棒性和灵活性[^4]。 --- ### 图解说明 以下是关于 Wise-IoU 的直观理解图示：  1. 左侧展示了标准 IoU 计算过程，仅仅依赖于交并比； 2. 右侧加入了 Wisely Weighting Mechanism 后的效果对比——可以看到红色箭头指向的小目标获得了更多注意资源，从而改善整体性能表现。 具体而言，图像中的颜色深浅反映了对应区域的重要性程度：越接近真实的 GT Box 边界处，颜色越浓烈，意味着此处受到更高强度的学习压力驱动优化方向。 ```python import torch def wiou_loss(pred_boxes, gt_boxes): """ Compute the Wise-IoU Loss between predicted boxes and ground truth boxes. Args: pred_boxes (Tensor): Predicted bounding boxes with shape [N, 4]. gt_boxes (Tensor): Ground-truth bounding boxes with shape [N, 4]. Returns: Tensor: Scalar tensor representing the loss value. """ # Calculate basic components of IoU x1_intersect = torch.max(pred_boxes[:, 0], gt_boxes[:, 0]) y1_intersect = torch.max(pred_boxes[:, 1], gt_boxes[:, 1]) x2_intersect = torch.min(pred_boxes[:, 2], gt_boxes[:, 2]) y2_intersect = torch.min(pred_boxes[:, 3], gt_boxes[:, 3]) intersection_area = torch.clamp(x2_intersect - x1_intersect, min=0) * \ torch.clamp(y2_intersect - y1_intersect, min=0) pred_area = (pred_boxes[:, 2]-pred_boxes[:, 0])*(pred_boxes[:, 3]-pred_boxes[:, 1]) gt_area = (gt_boxes[:, 2]-gt_boxes[:, 0])*(gt_boxes[:, 3]-gt_boxes[:, 1]) union_area = pred_area + gt_area - intersection_area iou = intersection_area / (union_area + 1e-7) # Define weights based on size or other criteria box_size_weights = 1 / ((pred_boxes[:, 2]-pred_boxes[:, 0]) * (pred_boxes[:, 3]-pred_boxes[:, 1])) weighted_iou = iou * box_size_weights return 1 - weighted_iou.mean() ``` --- ###