跳转至内容

PP-YOLOE+ 与YOLOv6.0:详细技术比较

在现代物体检测架构中,通常需要在针对特定框架生态系统进行优化的模型和针对原始工业速度设计的模型之间做出选择。本综合分析比较了来自PaddlePaddle 套件的高精度无锚检测器PP-YOLOE+ 和美团针对实时工业应用设计的以速度为中心的模型YOLOv6.0。通过研究它们的架构、性能指标和理想用例,开发人员可以确定哪种模型最符合他们的部署限制。

PP-YOLOE+:无锚精度

PP-YOLOE+是YOLO -YOLO系列的进化版。 PaddlePaddle生态系统的精度极限。PP-YOLOE+ 于 2022 年初发布,采用无锚设计,简化了训练流水线,同时为通用计算机视觉任务提供最先进的性能。

作者: PaddlePaddle 作者
组织:百度
日期:2022-04-02
ArXiv:https://arxiv.org/abs/2203.16250
GitHubPaddlePaddle
Docs:PaddlePaddle

架构与关键创新

PP-YOLOE+ 的架构建立在 CSPRepResNet骨干之上,它将残差网络的特征提取能力与跨阶段局部(CSP)连接的效率相结合。与传统检测器的一个显著不同之处在于它的无锚头,无需预定义锚框。超参数的减少简化了模型配置,提高了在不同数据集上的泛化能力。

最重要的是,PP-YOLOE+ 采用了任务对齐学习(TAL)技术来解决分类和定位任务之间的不对齐问题--这是单级检测器的常见问题。通过根据预测质量动态分配标签,TAL 可确保最高置信度分数与最准确的边界框相对应。

优势与劣势

优势:

  • 高精度:持续实现卓越的 mAP在COCO 等基准测试中,尤其是在较大的模型变体(如 PP-YOLOE+x)中,始终保持优异的 mAP 分数。
  • 简化训练:无锚范例消除了为确定锚大小而进行聚类分析的复杂性。
  • 生态系统协同效应:为已经使用PaddlePaddle 深度学习框架的用户提供深度集成。

弱点:

  • 推理延迟:与YOLOv6 等硬件感知模型相比,推理速度通常较慢,尤其是在GPU 硬件上。
  • 框架依赖性:与原生框架无关的架构相比,将模型移植到PyTorch 或ONNX 等其他框架进行部署可能会产生更多摩擦。

理想用例

在精度优先于超低延迟的情况下,PP-YOLOE+ 通常是首选。

  • 详细检测:检测生产质量控制中的微小缺陷,因为漏检一个故障的代价是高昂的。
  • 智能零售:用于货架监控和产品识别的高保真零售分析
  • 复杂分类:通过区分视觉上相似的材料来提高回收效率

了解更多关于 PP-YOLOE+ 的信息

YOLOv6.0:专为工业速度而设计

YOLOv6.0 由美团的视觉人工智能团队推出,旨在满足工业应用的严格要求。YOLOv6 优先考虑推理速度和准确性之间的权衡,采用了硬件感知设计原则,以最大限度地提高 GPU 和边缘设备的吞吐量。

作者:李楚一、李露露、耿一飞、蒋洪亮、程萌、张博、柯在丹、徐晓明、储祥祥李楚怡、李露露、耿一飞、蒋洪亮、程萌、张博、柯在丹、徐晓明、储祥祥
机构:美团网
日期:2023-01-13
ArXiv:https://arxiv.org/abs/2301.05586
GitHubYOLOv6
Docs:ultralytics

架构和主要特性

YOLOv6.0 采用了 "高效重参数化骨干 "技术,其灵感来自 RepVGG,该技术允许模型在训练期间具有复杂的结构以学习丰富的特征,但在推理期间具有简化的结构以提高速度。这种重新参数化技术是其实时推理能力的关键所在。

该模型还利用了自增强功能,即由较大的教师模型指导较小的学生模型的训练,从而在不增加运行时计算成本的情况下提高准确性。此外,YOLOv6 支持积极的模型量化,使其在计算资源有限的硬件上部署时非常有效。

移动优化

YOLOv6 包括一个专门针对移动 CPU 优化的 "Lite "系列模型,在无法使用GPU 加速的情况下,利用不同的区块来保持速度。

优势与劣势

优势:

  • 超凡速度:专为高吞吐量而设计,YOLOv6.0n 型号在 T4 GPU 上的延迟低于 2 毫秒。
  • 硬件优化:该架构对 TensorRT优化,最大限度地提高GPU 利用率。
  • 高效扩展:在计算成本(FLOPs)与精确度之间取得良好平衡。

弱点:

  • 任务范围有限:主要用于检测;缺乏对复杂任务的本地支持,如姿势估计 或定向边界旋转框检测)。
  • 社区支持:虽然有效,但与Ultralytics 模型相比,该生态系统在第三方集成和社区教程方面不够活跃。

理想用例

YOLOv6.0 在反应时间至关重要的环境中表现出色。

  • 机器人学:实现自主移动机器人(AMR)的导航和互动
  • 交通分析:实时交通管理系统需要对车辆进行即时计数和分类。
  • 生产线:用于包装分割和分拣的高速传送带监控。

了解有关YOLOv6的更多信息

性能对比

PP-YOLOE+ 注重准确性,而YOLOv6注重速度,这种设计理念上的差异在性能指标上清晰可见。PP-YOLOE+ 通常在模型复杂度的高端获得更高的mAP 分数,而YOLOv6 则在较小和较快模型的原始推理速度方面占据优势。

模型尺寸
(像素)
mAPval
50-95
速度
CPU ONNX
(毫秒)
速度
T4 TensorRT10
(毫秒)
参数
(M)
FLOPs
(B)
PP-YOLOE+t64039.9-2.844.8519.15
PP-YOLOE+s64043.7-2.627.9317.36
PP-YOLOE+m64049.8-5.5623.4349.91
PP-YOLOE+l64052.9-8.3652.2110.07
PP-YOLOE+x64054.7-14.398.42206.59
YOLOv6.0n64037.5-1.174.711.4
YOLOv6.0s64045.0-2.6618.545.3
YOLOv6.0m64050.0-5.2834.985.8
YOLOv6.0l64052.8-8.9559.6150.7

注:指标比较在很大程度上取决于所使用的特定硬件和导出格式(如ONNX 与TensorRT)。

数据表明,对于资源有限的边缘应用,YOLOv6.0n 在 FLOPs 和延迟方面提供了最低的进入门槛。相反,对于需要最大检测能力的服务器端应用,PP-YOLOE+x 提供了最高的精度上限。

Ultralytics 的优势:YOLO11

PP-YOLOE+ 和YOLOv6 则在各自的领域提供了强大的功能、 Ultralytics YOLO11提供了一个整体解决方案,在高精度和易用性之间架起了一座桥梁。YOLO11 不仅仅是一个模型,还是一个进入维护良好的生态系统的入口,旨在简化整个机器学习生命周期。

为什么选择Ultralytics?

  • 无与伦比的多功能性: YOLO11 主要是一个检测器,而YOLO11 则不同,它原生支持实例分割姿势估计 旋转框检测分类。这样,开发人员只需使用一个 API 就能解决多方面的计算机视觉问题。
  • 易用性: Ultralytics Python 软件包摒弃了复杂的模板代码。加载模型、运行推理和可视化结果只需三行代码即可完成。
  • 效率和内存 Ultralytics 模型针对高效训练进行了优化,与transformer架构(如 RT-DETR.
  • 生态系统支持:通过频繁的更新、丰富的文档以及用于无代码培训的Ultralytics HUB等工具,用户可以从与行业同步发展的平台中获益。

部署变得简单

Ultralytics 优先考虑可访问性。您可以立即运行高级推理:

from ultralytics import YOLO

# Load a pre-trained YOLO11 model
model = YOLO("yolo11n.pt")

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

# Display results
results[0].show()

这种简便性还延伸到部署方面,可通过单行导出功能将数据导出为以下格式 ONNX, OpenVINO和CoreML 等格式,确保您的模型在任何目标硬件上都能以最佳状态运行。

了解更多关于 YOLO11 的信息

结论

在 PP-YOLOE+ 和YOLOv6.0 之间做出选择,主要取决于项目的具体限制条件。PP-YOLOE+是PaddlePaddle 框架内要求高精度方案的有力竞争者,而YOLOv6.0则为严重依赖GPU 推断的工业环境提供了令人信服的速度优势。

不过,对于寻求多功能、面向未来的解决方案,同时兼顾最先进性能和开发人员体验的开发人员来说,Ultralytics 是一个不错的选择、 Ultralytics YOLO11仍然是最佳推荐。其广泛的任务支持、活跃的社区以及与现代 MLOps 工作流程的无缝集成,使其成为前沿视觉人工智能的标准。

其他模型对比

了解更多详细比较,找到适合您需求的机型:


评论