YOLOv10模型改进-损失函数改进-第89篇：YOLOv10改进策略【损失函数】| Huber Loss损失函数

一、本文介绍

本文记录的是利用Huber Loss改进YOLOv10的定位精度。Huber Loss在L1和L2 Loss之间取得平衡。

二、Huber Loss损失函数介绍

2.1 设计出发点

L1 Loss在误差接近零时梯度不连续，L2 Loss对异常值敏感，Huber Loss结合了两者的优点。

2.2 损失函数公式

Huber Loss：

huber(x) = 0.5 * x², if |x| <= δ huber(x) = δ * (|x| - 0.5 * δ), otherwise

其中：

• δ：超参数，控制L1和L2的切换点

三、Huber Loss的实现代码

importtorchimporttorch.nnasnnclass