资讯动态

072、YOLOv11改进-Shape-IoU损失函数替换CIoU即插即用涨点数学推导与实验

发布时间:2026/8/7 14:52:30 来源:尧图企业网站定制
072、YOLOv11改进-Shape-IoU损失函数替换CIoU即插即用涨点数学推导与实验踩坑实录:那个让我熬夜三天的边界框回归问题去年做工业缺陷检测项目,YOLOv11跑在钢材表面数据集上,mAP@0.5卡在87.3%死活上不去。调了三天学习率、数据增强、anchor配置,纹丝不动。最后翻论文发现CIoU在长宽比极端场景下的缺陷——它用反正切函数近似长宽比一致性,梯度传播时对窄长目标的回归几乎失效。换Shape-IoU后直接跳到89.1%,涨了1.8个点。今天就把这个坑填上。为什么CIoU不够“聪明”CIoU的损失函数长这样:L_CIoU = 1 - IoU + ρ²(b, b_gt)/c² + αv其中v = 4/π² * (arctan(w_gt/h_gt) - arctan(w/h))²,α是平衡系数。问题出在v的梯度上——当预测框和真实框的长宽比差异较大时,arctan函数的导数趋近于0,梯度消失。更致命的是,CIoU只考虑长宽比的一致性,忽略了形状本身的几何特性。想象一下检测电线杆(长宽比1:20),CIoU的v项几乎不提供有效梯度。Shape-IoU:让损失函数“看见”形状Shape-IoU的核心思想是把边界框的“形状”显式建模到损失函数中。数学上,它引入了一个形状惩罚项,直接度量预测框与真实框在宽度和高度上的绝对差异:L_Shape-Io

读完文章,也想定制专属网站?

尧图设计师 24 小时内与您沟通定制方案

免费获取报价