跳转至内容

YOLOX vs. YOLOv5:深入的架构与性能比较

选择正确的目标检测模型是决定任何计算机视觉项目成功的关键决策。本指南全面技术比较了AI领域的两个关键模型:旷视科技的YOLOX和Ultralytics YOLOv5。通过分析它们的架构、性能指标和训练生态系统,我们旨在帮助开发人员和研究人员为其特定的部署环境做出明智的选择。

模型介绍

这两种模型都出现在实时目标检测快速发展的时期,但它们采用了不同的架构理念来实现其性能。

YOLOX:一种无锚框方法

旷视科技的郑舸、刘松涛、王峰、李泽明和孙剑等研究人员于2021年7月18日发布的YOLOX,通过摒弃传统的锚框,引入了显著的转变。在其Arxiv技术报告中记载,YOLOX集成了无锚点设计、解耦头和SimOTA标签分配策略。这一设计旨在弥合学术研究与工业应用之间的鸿沟,在标准数据集上提供了强大的性能。

了解更多关于 YOLOX 的信息

YOLOv5:生产级视觉AI标准

由 Glenn Jocher 撰写并由 Ultralytics 于 2020 年 6 月 26 日发布,YOLOv5 迅速成为部署计算机视觉的行业标准。原生构建于 PyTorch 框架之上,它通过提供无与伦比的易用性、极快的训练速度和高度完善的代码库,使最先进的 AI 得以普及。YOLOv5 的架构专注于速度、准确性和部署便捷性的完美平衡,使其成为从边缘设备到大规模云部署的首选。

了解更多关于 YOLOv5

架构差异

理解这些网络的核心机制差异有助于阐明它们在不同任务中表现不同的原因。

无锚框与基于锚框

最显著的区别在于 YOLOX 的无锚框机制。像 YOLOv5 这样的传统模型依赖预定义的锚框来预测边界框,这需要在训练数据集上进行聚类分析以确定最佳锚框尺寸。YOLOX 消除了这一需求,直接在每个空间位置预测边界框坐标。尽管无锚框方法减少了设计参数和启发式调整的数量,但 YOLOv5 经过优化的基于锚框的方法,在其自动锚框功能的辅助下,确保了开箱即用的极其稳定和可预测的训练收敛。

解耦头与耦合头

YOLOX采用解耦头,意味着分类和回归任务被分离到不同的神经网络分支中。作者认为这解决了空间和语义特征学习之间的冲突。相反,YOLOv5(在其早期版本中)利用了高度优化的耦合头,最大限度地提高了计算效率并减少了推理延迟,这对于实时边缘计算至关重要。

架构演进

尽管YOLOX在2021年倡导了解耦头,但Ultralytics随后在YOLOv8和尖端的YOLO26等后续模型中采纳并完善了解耦架构,结合了两者的优点。

标签分配策略

YOLOX 利用 SimOTA 进行标签分配,将真实目标与预测的配对公式化为最优传输问题。这种动态分配改善了对拥挤场景的处理能力。YOLOv5 采用基于形状规则的鲁棒分配策略,确保高质量的正样本持续输入到损失函数中,这有助于其传奇般的训练稳定性。

性能与基准

速度和准确性之间的权衡是衡量这些架构的最终标准。下表展示了不同模型尺寸在标准基准上的性能。

模型尺寸
(像素)
mAPval
50-95
速度
CPU ONNX
(毫秒)
速度
T4 TensorRT10
(毫秒)
参数
(M)
FLOPs
(B)
YOLOXnano41625.8--0.911.08
YOLOXtiny41632.8--5.066.45
YOLOXs64040.5-2.569.026.8
YOLOXm64046.9-5.4325.373.8
YOLOXl64049.7-9.0454.2155.6
YOLOXx64051.1-16.199.1281.9
YOLOv5n64028.073.61.122.67.7
YOLOv5s64037.4120.71.929.124.0
YOLOv5m64045.4233.94.0325.164.2
YOLOv5l64049.0408.46.6153.2135.0
YOLOv5x64050.7763.211.8997.2246.4

尽管YOLOX取得了有竞争力的mAP分数,尤其是在其大型变体中,但YOLOv5在TensorRT推理速度方面全面保持显著优势。例如,YOLOv5s模型提供了卓越的速度-精度比,使其成为对每一毫秒都至关重要的实时应用中备受青睐的选择。

Ultralytics 优势:训练与可用性

从研究转向生产时,模型周围的生态系统通常与模型本身同等重要。在此,Ultralytics 生态系统的优势变得显而易见。

简化的用户体验

YOLOv5因其“从零到英雄”的开发者体验而广受赞誉。Ultralytics python API和CLI允许您通过单行代码加载、训练和部署模型。相比之下,从旷视GitHub仓库运行YOLOX需要更多手动配置环境变量、复杂的python路径设置以及学术研究代码库常见的陡峭学习曲线。

训练效率与内存需求

Ultralytics 模型经过精心设计,可在训练期间最大限度地减少内存使用。与RT-DETR等参数量大的 Transformer 模型或未优化的研究模型相比,YOLOv5 需要的 CUDA 内存显著更少。这使得开发者能够在消费级硬件上训练更大的批量大小,从而加速迭代开发周期。

跨任务多功能性

尽管YOLOX严格来说是一个目标检测框架,但Ultralytics生态系统已将YOLOv5发展为支持多种视觉任务。开箱即用,您可以使用完全相同的API语法执行图像分类实例segment和目标detect。

持续创新

如果您需要更高级的任务,例如姿势估计旋转框检测 (OBB),我们强烈建议升级到最新的Ultralytics YOLO26架构,该架构原生支持所有这些功能,并具有最先进的准确性。

代码对比

可用性差异通过代码演示最能体现。

使用 YOLOv5 进行训练:

from ultralytics import YOLO

# Load a pretrained YOLOv5s model
model = YOLO("yolov5su.pt")

# Run inference on an image
results = model("https://ultralytics.com/images/zidane.jpg")

# Display results
results[0].show()

使用 YOLOX 进行训练:(需要手动克隆仓库、安装 setup.py 和复杂的 CLI 参数)

# Example YOLOX training command
python tools/train.py -f exps/default/yolox_s.py -d 1 -b 64 --fp16 -o

Ultralytics 方法消除了障碍,让您能够专注于数据集和应用程序逻辑,而不是调试配置文件。此外,通过与 Weights & BiasesComet ML 的内置集成,您的实验跟踪将变得无缝。

理想用例和实际应用

在这些模型之间进行选择取决于您项目的运行环境。

YOLOX 的优势

YOLOX 在学术环境中仍然是一个强有力的候选者,尤其是在研究人员明确研究无锚框范式或标签分配策略时。它也适用于拥挤场景检测是绝对主要指标而边缘部署速度是次要考量的场景。

YOLOv5 的优势所在

YOLOv5是实际部署领域无可争议的冠军。

  • 高速制造:对于装配线缺陷检测,YOLOv5在边缘GPU上的极低推理延迟确保在不减慢传送带速度的情况下检查产品。
  • 无人机与航空影像:其高效的内存占用使其能够在无人机上的轻量级伴侣计算机上运行,执行诸如农业监测和野生动物追踪等任务。
  • 智慧零售:自动化结账 到库存管理,YOLOv5 可轻松导出到 TensorRTONNX,用于在数千个商店摄像头上进行大规模部署。

展望未来:YOLO26 的优势

尽管YOLOv5是一个传奇模型,但人工智能领域发展迅速。如果您今天开始一个新项目,我们强烈建议您关注Ultralytics的最新一代模型。

于2026年发布,Ultralytics YOLO26 代表着巨大的飞跃。它采用 端到端无 NMS 设计,完全消除了非极大值抑制后处理的需求,这极大地简化了部署逻辑。通过移除分布焦点损失(DFL)并利用尖端的 MuSGD 优化器,YOLO26 实现了比前几代快 43% 的 CPU 推理速度,同时保持更高的准确性,特别是对于小物体,这得益于新的 ProgLoss + STAL 损失函数。

无论您选择经过实战检验的 YOLOv5 可靠性,还是 YOLO26 的尖端性能,Ultralytics Platform 都能确保您拥有最佳工具,将您的计算机视觉解决方案从概念无缝推向生产。请务必查阅全面的Ultralytics 文档,以释放您的 AI 流水线的全部潜力。


评论