Press "Enter" to skip to content

YOLOX-PAI 详细解读(一)论文解读

本站内容均来自兴趣收集,如不慎侵害的您的相关权益,请留言告知,我们将尽快删除.谢谢.

YOLOX-PAI 论文解读

 

论文解读

 

2021 年,旷视提出 YOLOX 算法,在速度和精度上构建了新的基线,组件灵活可部署,深受工业界的喜爱。

 

近日,阿里云机器学习平台团队 PAI 通过自研的 PAI-EasyCV 框架复现 YOLOX 算法。

 

一、YOLOX-PAI是什幺?

 

YOLOX 是最着名的单阶段物体检测方法之一。

 

YOLOX-PAI是阿里在EasyCV中,利用 PAI 的推理优化框架 对YOLOX进行一系列优化得到的高效的预测器 api。

 

二、论文解读

 

我们开发了一个名为 EasyCV 的多功能计算机视觉工具箱,以方便使用各种 SOTA 计算机视觉方法。最近,我们将 YOLOX 的改进版 YOLOX-PAI 添加到 EasyCV 中。

 

通过消融研究以探讨某些检测方法对 YOLOX 的影响。我们还为 PAI-Blade 提供了一个简单的用法,用于加速基于 BladeDISCTensorRT 的推理过程。

 

最后,在 COCO 数据集上,单个 NVIDIA V100 GPU 上速度为1.0 毫秒,精度为42.8 mAP,这比 YOLOv6 快一点。

 

EasyCV 中还设计了一个简单但高效的预测器 api 来进行端到端目标检测。

 

YOLOX 是最着名的单阶段物体检测方法之一,已广泛应用于自动驾驶、缺陷检测等各个领域。它在 YOLO 系列中引入了解耦头和Anchor-free方式,并在 40 mAP 到 50 mAP 之间获得最先进的结果。

 

考虑到它的灵活性和效率,我们打算将 YOLOX 集成到 EasyCV 中,这是一种一体化的计算机视觉方法,即使是初学者也能轻松使用。此外,通过使用不同增强的检测Backbone、Neck和Head来研究YOLOX 的改进。用户可以根据自己的需求简单地设置不同的配置来获得适合自己的目标检测模型。

 

此外,基于 PAI-Blade(PAI 的推理优化框架),进一步加快了推理过程,并提供了一个简单的 api 来在 EasyCV 中使用 PAI-Blade。最后,设计了一个高效的预测器 api,以端到端的方式使用 YOLOX-PAI,大大加速了原始 YOLOX。YOLOX-PAI 与最先进的目标检测方法之间的比较如图 1 所示。

 

简而言之,我们的主要贡献如下:

EasyCV 中发布 YOLOX-PAI 作为一个简单而高效的目标检测工具(包含 docker 镜像、模型训练、模型评估和模型部署的过程)。我们希望即使是初学者也可以使用 YOLOX-PAI 来完成他的目标检测任务。
对现有的基于 YOLOX 的目标检测方法进行了消融研究,其中仅使用一个配置文件来构建自行设计的 YOLOX 模型。随着架构的改进和 PAI-Blade 的效率,模型在单个 NVIDIA Tesla V100 GPU 上进行推理,耗时在1ms ,获得了 40 mAP 和 50 mAP 中最先进的目标检测结果。
EasyCV 中提供了一个灵活的预测器 API ,分别加速了预处理、推理和后处理过程。这样,用户可以更好地使用 YOLOX-PAI 进行端到端的目标检测任务。

2.1、Backbone

 

最近, YOLOv6PP-YOLOE 已经将 CSPNetBackbone 替换为 RepVGG 。在RepVGG中,在推理过程中使用一个3×3的卷积块来代替多分支结构,这样既有利于节省推理时间,又有利于提高目标检测结果。 紧随 YOLOv6 其后,我们也使用基于 RepVGGBackbone

 

2.2、neck

 

我们使用两种方法来提升 YOLOXYOLOX-PAINeck 的性能:

 

 

    1. 用于特征增强的自适应空间特征融合(

ASFF

    1. )及其变体(记为

ASFF_Sim

    1. );

 

    1. GSConv,一个轻量级的卷积块,以降低计算成本。

 

 

原始的 ASFF 方法使用几个 vanilla 卷积块来首先统一不同特征图的维度。受 YOLOv5Focus 层的启发,我们通过使用非参数的切片操作和均值操作来替换卷积块以获得统一的特征图(表示为 ASFF_Sim )。具体来说, YOLOX 输出的每个特征图的操作在图 2 中定义。

 

我们还使用2种基于 GSConvNeck 来优化 YOLOX 。使用的 Neck 架构如图3和图4所示。两种架构的区别在于是否将所有块替换为 GSConv 。正如作者所证明的那样, GSconv 是专门为通道达到最大和尺寸达到最小的 Neck 设计的。

 

2.3、head

 

作者通过注意力机制增强了 YOLOX-Head ,以协调目标检测和分类的任务(表示为 TOOD-Head )。架构如图 5 所示。首先使用一个 Stem 层来进行通道压缩,接着通过堆叠卷积层得到中间的特征层。最后,根据不同的任务计算自适应权重。分别测试了在 TOOD-Head 中使用 vanilla 卷积和基于 repvgg 的卷积的结果。

 

 

2.4、PAI-Blade

 

PAI-Blade 是一个用于模型加速的简单且强大的推理优化框架。它基于许多优化技术,如 Blade Graph OptimizerTensorRTPAI-TAOTensor Accelerator and Optimizer )等。 PAI-Blade 将自动搜索优化输入模型的最佳方法。因此,没有模型部署专业知识的人也可以使用 PAI-Blade 来优化推理过程。作者在 EasyCV 中集成了 PAI-Blade 的使用,让用户只需更改导出配置即可获得高效的模型。

 

2.5、EasyCV Predictor

 

除了模型推断,预处理功能和后处理功能在端到端目标检测任务中也很重要,而现有的目标检测工具箱往往会忽略这些功能。在 EasyCV 中,我们允许用户灵活选择是否使用预处理/后处理程序导出模型。然后,提供了一个预测器 api 来执行高效的端到端目标检测任务,整个过程只需几行代码。

 

在本节中,我们将汇报上述方法在COCO数据集上的消融研究结果 ,并且和YOLOXPAI与SOTA对象检测方法进行比较。

 

3.1、SOTA对比

我们选择了在YOLOX-PAI中有用的改进,与SOTA YOLOv6方法在表1中进行了比较 。可以直观看出YOLOX-PAI 跟YOLOv6对应版本对比,快了很多。

具有更好的mAP(和 YOLOv6-tiny 相比0.2 mAP 和22%速度提升,与 YOLOv6-s 相比高了0.2 mAP和13%速度提升 )

 

3.2、消融研究

backbone的影响

如表1所示:YOLOX用一个 基于RepVGG 的骨干实现更好的mAP,只牺牲一点点速度。它确实增加了更多的参数和计算量,需要更多的计算资源,但不需要额外的推理时间。考虑到性能,我们在 EasyCV 中将其作为灵活的 配置设置 。

颈部的影响 。

ASFF和 ASFF_ Sim 的影响如表2所示。

 

与ASFF相比,ASFF_Sim只需要增加一些参数和计算量就能提高检测结果。然而,在推理时间上会加大,我们将在未来实施 CustomOP 来优化它。影响GSConv的结果如表3所示,GSConv将带来0.3 mAP的提升,并减少3%的速度,在单个NVIDIA V100 GPU上。

头部的影响。

表4显示了 TOODHead 的影响。

 

我们研究了不同数量的内部卷积层对模型的影响。

 

研究表明,当 添加额外的卷积层 时,检测结果会变得更好。选择合适的超参数可以很好权衡的是速度和精度之间。我们发现,使用vanilla卷积替换基于repconv的卷积层时,结果会变得更糟。当 堆叠的层数比较少 的时候,使用基于Repconv的 cls_conv/reg_conv 层,可以稍微改善结果。

 

3.3、端到端结果

 

表5显示了具有 不同导出配置 的YOLOXs模型端到端预测结果。这个表中的关键字与EasyCV中的配置文件相同。很明显, blade优化 有助于优化 推理过程 。此外, 预处理过程 可以通过导出的jit模型进行加速。至于 后处理 ,我们仍在努力实现更好CustomOP,可以通过PAI Blade进行优化来获得更好的提升。在图1的右侧,可以看出,通过优化 PAI-blade和EasyCV预测器 ,我们在可以在YOLOX上得到满意的端到端推理时间。

在本文中,我们介绍了YOLOX-PAI,它是基于EasyCV的YOLOX的改进版本。
通过 改进模型架构和PAI blade ,它在 40mAP到50mAP 中取得了最先进的目标检测结果。
我们还提供了一个 简单有效的预测api 来执行端到端目标检测。
EasyCV是一个专注于SOTA的多功能工具箱计算机视觉方法,特别是在 自监督学习和视觉转换器 。我们希望用户能够使用Easy CV来快速进行计算机视觉任务,爱上计算机视觉。

我们感谢所有在EasyCV中重新实现的算法的作者,以及他们在github社区的贡献。我们也感谢郭天佑、蒋娜娜、方海鹏、陈嘉宇和阿里巴巴PAI团队的许多其他成员他们的贡献,在建设YOLOX-PAI和EasyCV工具包上。

Be First to Comment

发表回复

您的电子邮箱地址不会被公开。