YOLOX: 让YOLO重回Anchor-Free, 成为最强YOLO

最新推荐文章于 2024-09-06 09:00:00 发布

AICVHub

最新推荐文章于 2024-09-06 09:00:00 发布

阅读量6.2k

点赞数 3

分类专栏：目标检测论文笔记文章标签： yolox anchor free end to end Decoupled head SimOTA

本文链接：https://blog.csdn.net/oYeZhou/article/details/119762459

版权

论文笔记同时被 2 个专栏收录

62 篇文章 27 订阅

订阅专栏

目标检测

17 篇文章 12 订阅

订阅专栏

YOLOX是YOLO系列的最新改进版，引入了Anchor-Free机制，结合DecoupledHead、强数据增强、SimOTA标签分配等技术，提升了检测性能并降低了复杂度。实验表明，YOLOX在保持快速的同时，达到了State-of-the-Art的性能。

摘要由CSDN通过智能技术生成

论文地址:https://arxiv.org/abs/2107.08430

1. 动机

yolo系列一直以来都是目标检测领域最火的方法. 从yolo2开始, yolo系列就一直是Anchor-based的检测器, 而近年来Anchor-free类的方法发展迅猛, 已经可以达到甚至超越Anchor-based的方法了. 但是yolo4和yolo5依然沿用了Anchor-based架构. 于是, 作者将Anchor-free引入到yolo中, 并使用了各种最新技术(如: decoupled head, SimOTA等), 从而形成了一个全新的yolo检测--YOLOX.

2. 方法

2.1. Baseline and backbone

考虑到yolo4和yolo5可能会在anchor-based pipline上过拟合, 以及yolo3的通用性, 于是作者采用yolo3-SPP作为yolox的baseline.

backbone方面,作者采用了3套:

yolo3中的DarkNet53, 用于和yolo3进行对比;
yolo5中的CSPNet, 用于达到较高的性能;
Tiny 和 Nano, 用于轻量化检测;

2.2. Decoupled Head

在目标检测领域, 分类和回归之间的矛盾是一个众所周知的问题. 解耦头(Decoupled Head)是一个很好的解决方案, 但yolo还未将此方案纳入其中. 作者设计了两个实验,验证了Decoupled Head能够提升yolo的性能:

1). 将yolo原有的head替换为Decoupled Head可以极大加速收敛;
2). Decoupled Head对于end2end是十分必要的(可降低性能下降的幅度);

于是,作者设计了一个轻量级的解耦头, 结构如图2所示:

从图2可以看出: 对于FPN输出的feature maps, 先经过一个1*1卷积层进行通道降维; 然后经过两个并行的分支, 每个分支有两个3*3卷积层;最后,再经过1*1卷积, 得到三个分支(Cls, Rge, IoU)。

2.3. Strong Data Augmentation

作者使用Mosaic和Mixup作为yolox中的数据增强策略, 并在最后15个epochs关闭。使用了这种强数据增强策略后，作者发现Imageet预训练就不再起作用了，因此，yolox是train from scratch的。

2.4. Anchor Free

yolo4和yolo5都延续了yolo3的anchor-based pipline，然而，anchor机制有很多已知的问题：

首先，为了达到最优性能，需要在训练前对数据集做聚类分析，来生成一组最优的anchors；这些anchors是特定于该数据集的，因此泛化性不佳；
其次，anchor机制增加了检测头的复杂度，以及每个图像中的预测数量；这对一些边缘AI设备不友好；

而anchor-free近些年发展迅猛，已经能够达到anchor-based方法的性能，且其显著减少了手工设计的内容以及各种tricks，从而使得训练和解码阶段得以简化。

将yolo切换到anchor-free也相当简单：

将每个位置预测个数从3减少到1，并直接预测四个值（即：到网格的左上角的偏移量和box的高宽）；同时，将中心点设为正样本，并预设了一个尺度范围为每个对象指定FPN级别。

如此一来，检测器的参数量和GFLOPS大大降低，速度更快，性能更好。

2.5. Multi Positives

上面的anchor-free版本为了和yolo3保持一致，只为每个对象分配一个正样本（中心点）。但其实其他高质量预测的位置同样又用，可以有利于梯度更新并缓解训练过程中正负样本不均衡的问题。这里采用了FCOS中的中心点采样方法，将中心点3*3邻域的位置都设为正样本。

2.6. SimOTA

先进的标签分配是近年来目标检测领域的另一个重大进展。基于旷视自家的OTA，作者总结了先进的标签分配的四个关键见解：1）损失/质量感知；2）中心点先验；3）每个ground-truth具有动态正anchor数量（dynamic top-k）；4）全局视野。OTA满足上面四条规则，因此作者选其作为标签分配策略。

具体的，OTA从一个全局视角分析了标签分配，并将分配过程表示为最优运输（Optimal Transport，OT）问题，在现有的分配策略中产生SOTA性能。然而，作者发现在使用Sinkhorn-Knopp算法解决OT问题时会带来25%的训练时间，这对于300个epochs的训练过程来讲是非昂贵。因此，作者将其简化为了dynamic top-k策略，命名为SimOTA，从而能得到近似解。

这里简要介绍SimOTA：