一、本文介绍
本文给大家带来的改进机制是更加聚焦的边界框损失Focaler-IoU以及我二次创新的InnerFocalerIoU同时本文的内容支持现阶段的百分之九十以上的IoU,比如Focaler-IoU、Focaler-MpdIoU、Innner-Focaler-MpdIoU、Inner-FocalerIoU包含非常全的损失函数,边界框的损失函数只看这一篇就够了。
欢迎大家订阅我的专栏一起学习YOLO!
专栏目录:YOLOv5改进有效涨点目录 | 包含卷积、主干、检测头、注意力机制、Neck上百种创新机制
专栏回顾:YOLOv5改进专栏——持续复现各种顶会内容——内含100+创新
二、Focaler-IoU原理
Focaler-IoU是一种在对象检测中用于边界框回归的损失函数。这种方法的基本原理可以从以下几个方面来理解:
1. 专注于不同的回归样本:Focaler-IoU 通过对不同的回归样本进行聚焦,来提高在不同检测任务中的探测器性能。这是通过线性区间映射来重构IoU损失,实现对不同样本的关注。
2. 解决困难和简单样本分布问题:它分析并考虑了困难样本和简单样本在边界框回归中的分布对回归结果的影响,这是传统IoU损失函数中常被忽视的一个方面。
3. 改进现有的边界框回归方法:Focaler-IoU 通过其特有的方法来弥补现有边界框回归方法的不足,从而在不同的检测任务中进一步提高检测性能。
上面这个公式定义了Focaler-IoU,它根据交并比(IoU)的值来调整损失。
当IoU小于一个下限阈值 d 时,损失为0;
当IoU大于一个上限阈值 u 时,损失为1;
而当IoU处于 d 和 u 之间时,损失是一个根据IoU值线性递增的函数。
这样的设计允许损失函数在一定区间内对IoU值敏感,从而能够更专注于那些预测边界框与真实边界框重叠度中等的样本,即既不是太难也不是太容易的样本。这有助于模型更好地学习从中等困难的样本中提取特征,而不是仅仅专注于最容易或最困难的样本。