YOLOv8损失函数篇 | 更换损失函数

📌损失函数篇 | YOLOv8更换损失函数之 SIoU、EIoU、WIoU、Focal_xIoU等

SloU

  • 论文地址SloU: More Powerful Learning for Bounding Box Regression
  • 提出时间
  • 适用场景:适用于需要高精度边界框对齐的场景,如精细的物体检测和小目标检测。
  • 概念:SloU损失通过融入角度考虑和规模敏感性,引入了一种更为复杂的边界框回归方法,解决了以往损失函数的局限性,SloU损失函数包含四个组成部分:角度损失、距离损失、形状损失和第四个未指定的组成部分。通过整合这些方面,从而实现更好的训练速度和预测准确性。

WIoU

GloU

DloU

EloU

  • 论文地址Focal and Efficient IOU Loss for Accurate Bounding Box Regression
  • 提出时间:中科院提出
  • 适用场景:可用于需要进一步优化边界框对齐和形状相似性的高级场景。
  • 概念:EloU损失函数的核心思想在于提高边界框回归的准确性和效率。它通过以下几个方面来优化目标检测:
  1. 增加中心点距离损失:通过最小化预测框和真实框中心点之间的距离,提高边界框的定位准确性。
  2. 考虑尺寸差异:通过惩罚宽度和高度的差异,EloU确保预测框在形状上更接近真实框。
  3. 结合最小封闭框尺寸:将损失函数与包含预测框和真实框的最小封闭框的尺寸相结合,从而使得损失更加敏感于对象的尺寸和位置。

EloU损失函数在传统loU基础上增加了这些考量,以期在各种尺度上都能获得更精确的目标定位,尤其是在物体大小和形状变化较大的场景中。

Inner-IoU

  • 论文地址Inner-IoU: More Effective Intersection over Union Loss with Auxiliary Bounding Box
  • 提出时间:2023
  • 动机:提出了Inner-IoU损失,通过使用辅助边界框来加速回归,而无需添加任何新的损失项。其通过尺度因子ratio控制辅助边框尺寸以计算损失加速收敛,并能够集成至现有基于IoU的损失函数GIoU、DIoU、CIoU、EIoU、SIoU中。
  • 引入比例因子控制来生成不同尺度的辅助边界框以计算损失。

自己的数据集效果不是很好。

Focal loss

  • 论文地址Focal Loss for Dense Object Detection
  • 提出时间: Facebook AI团队
  • 概念:提出目标便是解决高低质量样本类别不平衡的问题。
  • 适用场景:适用于需要高精度边界框对齐的场景,如精细的物体检测和小目标检测。

使用Focal_xIoU类损失一般都会提升精度。


整理不易🚀🚀,关注和收藏后拿走📌📌欢迎留言🧐👋📣✨
快来关注我的公众号🔎AdaCoding 和 GitHub🔎 AdaCoding123
在这里插入图片描述

  • 11
    点赞
  • 56
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
YOLOv8中,损失函数是通过计算预测框与真实框之间的差异来衡量模型的性能。YOLOv8使用的损失函数是基于均方误差(Mean Squared Error)和交叉熵(Cross Entropy)的组合。 具体来说,YOLOv8损失函数由以下几个部分组成: 1. 边界框损失(Box Loss):用于衡量预测框与真实框之间的位置差异。YOLOv8使用均方误差作为边界框损失函数,计算预测框的中心坐标、宽度和高度与真实框的中心坐标、宽度和高度之间的差异。 2. 目标分类损失(Object Classification Loss):用于衡量预测框是否包含目标物体的准确性。YOLOv8使用交叉熵作为目标分类损失函数,计算预测框中每个类别的概率与真实框中的类别之间的差异。 3. 无目标分类损失(No-Object Classification Loss):用于衡量预测框是否不包含目标物体的准确性。YOLOv8同样使用交叉熵作为无目标分类损失函数,计算预测框中每个类别的概率与0(表示无目标)之间的差异。 4. 框置信度损失(Box Confidence Loss):用于衡量预测框与真实框之间的置信度差异。YOLOv8使用均方误差作为框置信度损失函数,计算预测框的置信度与真实框的置信度之间的差异。 这些损失函数的计算结果会被加权求和,得到最终的损失值。通过最小化损失函数YOLOv8可以优化模型的参数,使其能够更准确地检测和定位目标物体。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值