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