
一、本文介绍本文记录的是利用Focal Loss改进YOLOv10的分类精度。Focal Loss通过降低易分样本的权重,聚焦难分样本。二、Focal Loss损失函数介绍2.1 设计出发点传统交叉熵损失对所有样本同等对待,导致训练过程中大量易分样本主导梯度,Focal Loss通过引入调制因子解决这个问题。2.2 损失函数公式Focal Loss:FL(pt) = -α_t * (1 - pt)^γ * log(pt)其中:pt:预测概率α_t:平衡因子γ:聚焦参数三、Focal Loss的实现代码importtorchimporttorch.nnasnnclass