yolov5使用最新MPDIOU损失函数,有效和准确的边界盒回归的损失,优于GIoU/EIoU/CIoU/EIoU(附代码可用)

MPDIoU是一种新的边界框回归损失函数,旨在解决现有损失函数在相同纵横比但不同尺寸边界框上的局限。通过最小化左上角和右下角点的距离,它提高了效率和精度,特别适用于长宽比相同的预测框。MPDIoU计算简单,考虑了重叠、中心点距离、宽度和高度偏差。在实践中,MPDIoU损失函数可提升模型训练精度,特别是在处理包含关系的预测框时。

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

1. 论文

MPDIoU: A Loss for Efficient and Accurate Bounding Box Regression
(一个有效和准确的边界框损失回归函数)
论文地址

1.1. 主要目的

当预测框与边界框具有相同的纵横比,但宽度和高度值完全不同时,大多数现有的边界框回归损失函数都无法优化。
为了解决上述问题,MPDIoU充分挖掘水平矩形的几何特征,提出了一种新的基于最小点距离的边界框相似性比较指标MPDIoU,该指标包含了现有损失函数中考虑的所有相关因素,即重叠或不重叠区域、中心点距离、宽度和高度偏差,同时简化了计算过程。
在这里插入图片描述

论文中举例了,绿色框为真实目标框,红色框表示预测目标框,两幅图中预测目标框和真实目标框存在相同的长宽比,但是现象是不一致的,一个在真实框内部一个在真实框外部。

此时GIo

### YOLOv10 损失函数改进方法研究 #### 多种IoU损失函数及其变体的应用 YOLOv10引入了多种基于交并比(IoU)损失函数以及它们的不同变体,如SIoU、WIoU、GIoU、DIoU、EIOUCIoU等。这些改进后的损失函数不仅提高了模型在目标检测任务中的精度,也加速了训练过程中的收敛速度[^1]。 #### “Focus”思想与新型损失函数的设计 除了传统的IoU损失外,“Focus”概念被融入到了新的损失函数设计之中。“Focus”的核心在于更加关注那些难以区分的对象边界区域,在此基础上衍生出了超过二十种不同的组合型损失函数,每一种都针对特定类型的物体识别难题进行了优化调整。 #### MPDIoU InnerMPDIoU 的提出与发展 最新版本中提出了两种全新的损失函数——MPDIoUInnerMPDIoU。特别是后者作为首次公开的技术方案,结合了最新的“Inner”理念,在大多数情况下展现出优于其他现有解决方案的表现。这两种损失函数特别强调对于细粒度特征的学习能力,有助于进一步提升YOLOv10的整体表现水平[^2]。 #### EfficiCLoss 及 EIOU_Loss 对传统 CIOU_Loss 的超越 EfficiCLoss作为一种高效的替代选项也被纳入考虑范围之内;而EIOU_Loss则是在经典CIOU_Loss基础上做出的重要革新,主要体现在对长宽比例影响因素更为精细地建模上,使得预测框与真实标签之间的匹配程度得到显著改善[^3]。 ```python def compute_loss(pred_boxes, true_boxes): """ 计算不同种类的 IoU Loss 参数: pred_boxes (Tensor): 预测框坐标张量 true_boxes (Tensor): 真实框坐标张量 返回: Tensor: 各类 IoU loss 值之 """ iou_losses = [] # 实现 SIoU, WIoU, GIoU, DIoU, EIou, CIoU... siou_loss = calculate_siou_loss(pred_boxes, true_boxes) wiou_loss = calculate_wiou_loss(pred_boxes, true_boxes) giou_loss = calculate_giou_loss(pred_boxes, true_boxes) diou_loss = calculate_diou_loss(pred_boxes, true_boxes) eiou_loss = calculate_eiou_loss(pred_boxes, true_boxes) ciou_loss = calculate_ciou_loss(pred_boxes, true_boxes) iou_losses.extend([siou_loss, wiou_loss, giou_loss, diou_loss, eiou_loss, ciou_loss]) total_iou_loss = sum(iou_losses) return total_iou_loss ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值