ARTICLE DETAIL

资讯详情

深耕编程入门与网站建设的一线实战洞察。

074、YOLOv11改进-Focaler-IoU损失函数即插即用涨点数学推导与调优

074、YOLOv11改进-Focaler-IoU损失函数即插即用涨点数学推导与调优 074、YOLOv11改进-Focaler-IoU损失函数即插即用涨点数学推导与调优上个月调一个工业缺陷检测模型,YOLOv11s在钢带表面划痕数据集上跑了三天,mAP@0.5卡在78.3%死活上不去。可视化预测结果发现,模型对细长条划痕的定位框总是偏大一圈,召回率看着还行,精确率被拖累得厉害。换了几种数据增强策略都没用,直到把CIoU换成Focaler-IoU,mAP直接跳到81.1%,推理速度几乎没变。今天就把这个即插即用的涨点技巧拆开揉碎讲清楚。为什么CIoU在细长目标上会失效CIoU的损失函数长这样:L_CIoU = 1 - IoU + (ρ²(b, b_gt) / c²) + αv其中v是宽高比一致性度量,α是平衡系数。问题出在v这个项上——它只惩罚宽高比不一致,不惩罚宽高本身的大小差异。对于细长目标,预测框和真实框的宽高比可能很接近,但实际尺寸差了一截。CIoU的梯度信号在这种情况下会变得很弱,模型学不动。更致命的是,当预测框和真实框完全不重叠时,IoU=0,梯度直接消失。这就是为什么早期训练阶段模型收敛慢,后期对难样本定位不准。Focaler-IoU的核心思路Focaler-IoU的数学形式借鉴了Focal Loss的思想,把关注点从"所有样本一视同仁"转向"聚焦困难样本"。它的损失函数定义如下:L_FocalerIoU = (1 - IoU)^γ * L_IoU_base
返回列表