摘要:
YOLOv5是一种流行的目标检测算法,具有高效的推理速度和较好的性能。然而,传统的损失函数在训练过程中无法很好地平衡不同类型的目标,导致对小目标和困难样本的检测性能不佳。为了解决这个问题,我们提出了一种改进的损失函数,结合了多种类别的损失函数,包括PolyLoss、VarifocalLoss、GFL、QualityFLoss和FocalLoss等。通过实验验证,我们的方法在多个常用数据集上均取得了显著的性能提升。
-
引言
目标检测是计算机视觉领域的重要任务之一,其在自动驾驶、视频监控、智能安防等领域有着广泛的应用。YOLOv5作为目标检测领域的先进算法,以其高效的推理速度和较好的性能受到了广泛关注。然而,对于小目标和困难样本的检测,传统的YOLOv5模型存在一定的局限性。 -
相关工作
</
为了解决目标检测中类别不平衡和困难样本带来的问题,研究者们提出了许多不同的损失函数。PolyLoss是一种通过添加多项式系数来加权损失函数的方法,可以增强对小目标的检测能力。VarifocalLoss引入了一个可变的焦点权重来平衡正负样本的梯度,提升了困难样本的检测性能。GFL则是通过引入高斯分布来建模目标尺寸,提高了对小目标的检测能力。QualityFLoss通过引入质量因子来平衡正负样本的难易程度,改善了类别不平衡问题。FocalLoss是一种常用的解决类别不平衡问题的损失函数,通过减小易分类样本的权重来关注困难样本。