YOLOX 与 YOLOv10:无锚框(Anchor-Free)与无需非极大值抑制(NMS-Free)实时目标检测的对比

实时计算机视觉模型的演进标志着架构上的重大飞跃。YOLOX 和 YOLOv10 是这一进程中的两个关键里程碑。YOLOX 发布于 2021 年,通过引入高效的无锚框设计,成功架起了学术研究与工业应用之间的桥梁。三年后,YOLOv10 通过在后处理阶段消除对非极大值抑制(NMS)的需求,彻底革新了该领域,突破了效率与速度的界限。

本综合技术对比探讨了两种模型的架构、性能指标及理想用例,旨在为你选择下一个目标检测项目的最佳工具提供参考。

模型来源与元数据

了解这些模型的起源有助于掌握其架构选择及预期部署环境的背景。

YOLOX 详情 作者:Zheng Ge, Songtao Liu, Feng Wang, Zeming Li, and Jian Sun 组织:Megvii 日期:2021-07-18 Arxiv: https://arxiv.org/abs/2107.08430 GitHub: https://github.com/Megvii-BaseDetection/YOLOX 文档:https://yolox.readthedocs.io/en/latest/

了解关于 YOLOX 的更多信息

YOLOv10 详情 作者:Ao Wang, Hui Chen, Lihao Liu, Kai Chen, Zijia Lin, Jungong Han, and Guiguang Ding 组织:Tsinghua University 日期:2024-05-23 Arxiv: https://arxiv.org/abs/2405.14458 GitHub: https://github.com/THU-MIG/yolov10 文档:https://docs.ultralytics.com/models/yolov10/

了解关于 YOLOv10 的更多信息

架构创新

YOLOX 与 YOLOv10 的核心差异在于它们处理边界框预测和后处理的方式。

YOLOX:无锚框设计的先驱

YOLOX 通过将 YOLO 系列转向无锚框架构,引发了巨大反响。它通过预测物体中心而非依赖预定义的锚框,大幅减少了自定义数据集所需的设计参数和启发式调优工作。此外,它引入了解耦头(decoupled head),将分类和回归任务分离为独立的路径。这种方法解决了识别物体是什么与确定物体在哪里之间的冲突,从而显著提升了收敛速度和精度。

YOLOv10:无需 NMS 的革命

尽管 YOLOX 简化了检测头,但它仍依赖 NMS 来过滤冗余的边界框预测。YOLOv10 解决了这一根本瓶颈。通过在训练过程中使用一致的双重分配(consistent dual assignments),YOLOv10 实现了原生端到端检测。它在训练时采用多对一(one-to-many)检测头以确保丰富的监督信号,而在推理时则利用一对一(one-to-one)检测头直接输出最终预测。这种基于效率与精度权衡的整体设计完全消除了 NMS,显著降低了嵌入式芯片上的推理延迟。

移除 NMS 的影响

非极大值抑制(NMS)通常是在神经处理单元(NPU)上加速的一项复杂操作。通过移除它,YOLOv10 允许整个模型图在专用硬件上无缝执行,从而极大地提高了与 OpenVINO 和 TensorRT 等优化框架的兼容性。

性能指标与比较

在评估生产环境模型时,平衡精度与计算开销至关重要。下表展示了不同规模 YOLOX 与 YOLOv10 模型之间的权衡。

模型尺寸
(像素)
mAPval
50-95
速度
CPU ONNX
(ms)
速度
T4 TensorRT10
(ms)
参数
(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
YOLOv10n64039.5-1.562.36.7
YOLOv10s64046.7-2.667.221.6
YOLOv10m64051.3-5.4815.459.1
YOLOv10b64052.7-6.5424.492.0
YOLOv10l64053.3-8.3329.5120.3
YOLOv10x64054.4-12.256.9160.4

分析数据

这些指标清楚地证明了 YOLOv10 的代际飞跃。例如,YOLOv10-S 的 mean Average Precision 达到 46.7%,与 YOLOX-m 的 46.9% 相当,但其参数量不到后者的三分之一(7.2M 对比 25.3M),且 FLOPs 显著更少。此外,顶级的 YOLOv10-X 模型将 mAP 推高至 54.4%,在应对高精度需求任务时极具竞争力,同时比旧的 YOLOX-x 架构更快。

Ultralytics 生态系统的优势

虽然 YOLOX 仍然是一个强大的开源研究实现,但采用 YOLOv10 可以让你直接获得由 Ultralytics 提供的维护良好的生态系统。选择 Ultralytics 支持的模型可确保获得由简洁 API 和详尽文档支持的精简用户体验。

开发者可从该框架的内存需求中获益良多;训练 Ultralytics 模型通常比 RT-DETR 等沉重的基于 Transformer 的替代方案消耗更少的 CUDA 内存。这种高效的训练占用空间允许在消费级硬件上使用更大的批处理大小,从而加快从数据收集到模型部署的进程。此外,该框架提供了无与伦比的多功能性,允许用户在 object detectioninstance segmentationpose estimation 之间轻松切换,只需极少的代码更改。

训练与推理示例

统一的 API 使得验证想法变得非常迅速。以下代码片段演示了如何使用 PyTorch 后端轻松训练和部署 YOLOv10 模型:

from ultralytics import YOLO

# Load a pre-trained YOLOv10 nano model
model = YOLO("yolov10n.pt")

# Train the model on the COCO8 dataset
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)

# Run inference on a sample image
predictions = model.predict("https://ultralytics.com/images/bus.jpg")

# Export the model for edge deployment
model.export(format="engine", half=True)

通过利用内置的导出例程,将模型转换为 TensorRTONNX 等格式仅需一行代码,从而完全绕过复杂的编译障碍。

理想的使用场景和部署方案

在这些架构之间进行选择,主要取决于你的硬件限制和特定领域的具体需求。

实时视频分析

对于需要超低延迟的应用程序,例如自动驾驶或实时交通监控,YOLOv10 是首选。其端到端的 NMS-Free 设计保证了确定性的执行时间,这对于无法容忍可变后处理延迟的安全系统至关重要。这些模型可以轻松在 NVIDIA Jetson 系列等设备上实现高帧率。

学术基准与边缘微控制器

YOLOX 在学术界仍然很有价值,研究人员希望使用一个干净、具备解耦检测头的基准来实验标签分配策略。此外,极其小巧的 YOLOX-Nano(参数少于 100 万)可以塞进内存仅以千字节计的高度受限的边缘微控制器中,前提是硬件支持标准的卷积运算。

终极标准:Ultralytics YOLO26

虽然 YOLOv10 通过移除 NMS 实现了巨大的飞跃,但计算机视觉领域发展迅速。对于旨在实现当今一流性能的开发者,我们强烈建议探索 YOLO26

作为视觉 AI 的最新标准,YOLO26 汲取了前代产品的基本理念并进行了全面强化。它提供了极致的性能平衡,原生支持检测、分割、姿态估计和旋转边界框(Oriented Bounding Boxes)。

以下是为何 YOLO26 成为现代计算机视觉流水线推荐选择的原因:

  • 端到端 NMS-Free 设计: 在 YOLOv10 的突破基础上,YOLO26 原生支持端到端,确保了更快、确定性的推理时间,消除了后处理瓶颈。
  • CPU 推理速度提升高达 43%: 它专为边缘计算进行了优化,确保在移动处理器和缺乏独立 GPU 的设备上具有出色的性能。
  • MuSGD 优化器: 受大语言模型训练(特别是 Moonshot AI 的 Kimi K2)的启发,YOLO26 采用 SGD 和 Muon 的混合优化方案,以实现极其稳定的训练和快速收敛。
  • ProgLoss + STAL: 这些先进的损失函数在小目标识别方面提供了显著改进,这对航空影像和无人机导航等高要求领域至关重要。
  • 移除 DFL: 通过移除分布焦点损失(Distribution Focal Loss),YOLO26 简化了模型图,实现了向边缘和低功耗设备无摩擦的导出。
  • 任务特定改进: 无论你是使用残差对数似然估计(RLE)进行姿态估计,还是使用专用的角度损失处理 OBB,YOLO26 都针对每个主要视觉任务进行了微调。

对于准备好使用现有最高效的训练和部署工具升级流水线的开发者而言,迁移到 Ultralytics Platform 并利用 YOLO26 可确保你始终站在人工智能的最前沿。对较旧但稳定架构感兴趣的用户也可以查阅 YOLO11YOLOv8,以获得广泛的社区支持和经过验证的稳健性。

评论