EfficientDet 与YOLOv6.0:综合技术比较
在不断发展的计算机视觉领域,选择正确的物体检测架构对于成功部署至关重要。本比较探讨了Google的研究型EfficientDet 和美团的工业级检测器YOLOv6.0 之间的技术区别。EfficientDet 引入了复合缩放等突破性的效率概念,而YOLOv6.0 则是专为低延迟工业应用而设计,突出了从学术基准到实际吞吐量的转变。
性能指标比较
以下有关COCO 数据集的基准测试说明了架构效率与推理延迟之间的权衡。YOLOv6.0 利用重新参数化技术,在GPU 硬件上展示了卓越的速度,而 EfficientDet 则以较高的计算成本保持了具有竞争力的精确度。
| 模型 | 尺寸 (像素) | mAPval 50-95 | 速度 CPU ONNX (毫秒) | 速度 T4 TensorRT10 (毫秒) | 参数 (M) | FLOPs (B) |
|---|---|---|---|---|---|---|
| EfficientDet-d0 | 640 | 34.6 | 10.2 | 3.92 | 3.9 | 2.54 |
| EfficientDet-d1 | 640 | 40.5 | 13.5 | 7.31 | 6.6 | 6.1 |
| EfficientDet-d2 | 640 | 43.0 | 17.7 | 10.92 | 8.1 | 11.0 |
| EfficientDet-d3 | 640 | 47.5 | 28.0 | 19.59 | 12.0 | 24.9 |
| EfficientDet-d4 | 640 | 49.7 | 42.8 | 33.55 | 20.7 | 55.2 |
| EfficientDet-d5 | 640 | 51.5 | 72.5 | 67.86 | 33.7 | 130.0 |
| EfficientDet-d6 | 640 | 52.6 | 92.8 | 89.29 | 51.9 | 226.0 |
| EfficientDet-d7 | 640 | 53.7 | 122.0 | 128.07 | 51.9 | 325.0 |
| YOLOv6.0n | 640 | 37.5 | - | 1.17 | 4.7 | 11.4 |
| YOLOv6.0s | 640 | 45.0 | - | 2.66 | 18.5 | 45.3 |
| YOLOv6.0m | 640 | 50.0 | - | 5.28 | 34.9 | 85.8 |
| YOLOv6.0l | 640 | 52.8 | - | 8.95 | 59.6 | 150.7 |
EfficientDet:可扩展效率
通过系统地优化网络深度、宽度和分辨率,EfficientDet 实现了模型设计的范式转变。它以 EfficientNet 为骨干,引入了双向特征金字塔网络(BiFPN),可轻松实现多尺度特征融合。
- 作者: Mingxing Tan、Ruoming Pang 和 Quoc V. Le谭明星、庞若明和 Quoc V. Le
- 组织机构Google
- 日期: 2019-11-20
- Arxiv: https://arxiv.org/abs/1911.09070
- GitHub: https:google
- 文件 :google
建筑创新
EfficientDet 的核心是BiFPN,它允许信息自上而下和自下而上地流动,反复融合不同尺度的特征。这与老式检测器经常使用的简单特征金字塔网络(FPN)形成了鲜明对比。此外,EfficientDet 还采用了复合缩放法(Compound Scaling),这是一种使用单一复合系数 $\phi$ 对骨干网、BiFPN 和类/盒网络进行统一缩放的方法。这种结构化的方法确保了模型各维度的资源平衡,避免了人工设计架构中经常出现的瓶颈问题。
优势与劣势
EfficientDet 在参数效率方面表现出色,可实现较高的 mAP与 YOLOv3 等同类产品相比,它能以相对较少的参数实现较高的 mAP。在模型大小(存储)受限但延迟可协商的图像分类和检测任务中,它尤其有效。不过,BiFPN 层中复杂的不规则连接和对深度可分离卷积的广泛使用可能会降低标准 GPU 的效率,从而导致尽管 FLOP 数较低,但推理延迟较高。
延迟与 FLOP
虽然 EfficientDet 的 FLOPs(浮点运算)较低,但在 GPU 上并不总能转化为更快的速度。与YOLO 模型中使用的标准卷积相比,YOLO 深度可分离卷积的内存访问成本可能会成为性能瓶颈。
YOLOv6.0:工业速度
YOLOv6.0 摆脱了纯粹的学术指标,转而关注实际吞吐量,特别是针对工业环境中的硬件限制进行优化。
- 作者: Chuyi Li, Lulu Li, Yifei Geng, Hongliang Jiang, Meng Cheng, Bo Zhang, Zaidan Ke, Xiaoming Xu, 和 Xiangxiang Chu
- 组织: 美团
- 日期: 2023-01-13
- Arxiv: https://arxiv.org/abs/2301.05586
- GitHub: https:YOLOv6
- 文件 :ultralytics
架构与设计
YOLOv6.0 采用了EfficientRep Backbone,它利用重参数化(RepVGG 风格)将训练时间和推理时间架构解耦。在训练过程中,模型使用复杂的多分支块来获得更好的梯度流;在推理过程中,这些块折叠成单次 3 美元的卷积,从而最大化 GPU计算密度。3.0 版还集成了量化感知训练(QAT)和自蒸馏等先进策略,使模型即使量化到 INT8 精度也能保持精度,以便在边缘设备上部署。
理想用例
由于YOLOv6.0 采用了硬件友好型设计,因此非常适合以下应用:
比较分析
这两种模式在设计理念上的差异会根据部署硬件的不同而产生不同的优势。
精度与速度
如表所示,YOLOv6.0l的mAP (52.8)与EfficientDet-d6的mAP (52.6)相当,但在 T4GPU 上的运行速度却快了近10 倍(8.95ms 对 89.29ms)。这一巨大差距凸显了深度卷积在高吞吐量硬件上的效率低于YOLOv6 的稠密卷积。EfficientDet 以其最大的 D7 变体在绝对精度上略胜一筹,但其延迟成本导致无法进行实时推理。
培训和多功能性
EfficientDet 在很大程度上依赖于TensorFlow 生态系统和TPU 加速来实现高效训练。相比之下,YOLOv6 适用于PyTorch 生态系统,使普通研究人员更容易使用。不过,这两种模型都主要用于对象检测。对于需要实例分割或姿势估计 项目,用户通常需要寻找外部分叉或替代架构。
Ultralytics 的优势
而YOLOv6.0 和 EfficientDet 是有能力的模型、 Ultralytics YOLO11代表了计算机视觉领域的新发展,通过一个统一的、以用户为中心的框架解决了前两者的局限性。
为什么选择 Ultralytics YOLO11?
- 易用性与生态系统:与分散的研究模型库不同,Ultralytics 提供的是一种无缝体验。一致的Python 应用程序接口让您只需几行代码就能训练、验证和部署模型。
- 无与伦比的多功能性:YOLO11 不局限于边界框。它原生支持 图像分类, 实例分割, 姿势估计和 旋转框检测Oriented Bounding Boxes)使其成为复杂人工智能管道的一站式解决方案。
- 训练效率:Ultralytics 模型针对内存需求进行了优化,通常收敛速度更快,使用的 VRAM 也比transformer或老式架构更少。对于没有大规模计算集群的人来说,这种可访问性使高端人工智能开发平民化。
- 维护良好的生态系统:Ultralytics 生态系统拥有活跃的社区和频繁的更新支持,可确保您的项目面向未来,并可轻松集成到数据注释、日志记录和部署工具中。
简化开发
使用Ultralytics,从对象检测切换到实例分割只需更改模型名称(例如......)、 yolo11n.pt 到 yolo11n-seg.pt).与针对新任务调整不同的架构(如 EfficientDet)相比,这种灵活性大大缩短了开发时间。
代码示例
与复杂的研究代码库相比,体验Ultralytics API 的简易性:
from ultralytics import YOLO
# Load a pre-trained YOLO11 model
model = YOLO("yolo11n.pt")
# Train the model on your custom dataset
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)
# Run inference on an image
results = model.predict("https://ultralytics.com/images/bus.jpg")
结论
EfficientDet 仍然是模型扩展理论的里程碑,非常适合以准确性为唯一标准的学术研究或离线处理。YOLOv6.0 推动了工业边缘人工智能的发展,在支持的硬件上提供了卓越的速度。
然而,要想获得兼顾最先进性能和开发人员工作效率的整体解决方案,Ultralytics 是一个不错的选择、 Ultralytics YOLO11是值得推荐的选择。它集成了各种视觉任务,内存占用更低,并拥有强大的支持系统,使开发人员能够满怀信心地从原型转向生产。
探索其他模型
如果您有兴趣进一步了解,请参考我们文档中的相关比较: