遇见 YOLO26: 下一代视觉 AI。
Ultralytics
Ultralytics YOLO

YOLOv5 在 v6.1 中变得更强大了!

探索 Ultralytics 推出的 YOLOv5 v6.1,获取视觉 AI 的尖端增强功能,包含对 TensorRT、TensorFlow Edge TPU 的支持等。

ULUltralytics 团队
3 min read
Ultralytics YOLOv5 v6.1 版本改进

YOLOv5 v6.1 release

作为计算机视觉和机器学习领域的先驱,Ultralytics 很高兴宣布我们在旗舰 YOLO (You Only Look Once) 技术上的最新进展。随着 YOLOv5 v6.1 版本的发布,我们对架构进行了微调,以增强其简洁性、速度和强度,确保我们的技术始终处于创新的前沿。我们之前的发布(2021 年 10 月)为这些进步奠定了基础,现在我们很自豪地推出这些重新定义 YOLO 可用性和性能的关键更新。

Link to this section重要更新#

为了继续在 Vision AI 领域不懈追求卓越,以下是你将在 YOLOv5 v6.1 中发现的突破性增强功能:

  • TensorRT 支持:改进了使用 python export.py --include saved_model pb tflite tfjs 进行 TensorFlow、Keras、TFLite 和 TF.js 模型导出的集成 (#5699,由 @imyhxy 提供)。这是一个重要的里程碑,因为 NVIDIA 的 TensorRT 是一个高性能深度学习推理优化器和运行时,可为深度学习应用提供低延迟、高吞吐量的支持。
  • TensorFlow Edge TPU 支持 ⭐ 新增:推出了全新的更小巧的 YOLOv5n (1.9M 参数) 模型,其复杂度低于 YOLOv5s (7.5M 参数),但在导出为仅 2.1 MB 的 INT8 大小时表现出色。这对于超轻量级移动解决方案尤其理想,将强大的机器学习带到了技术的边缘 (#3630,由 @zldrobit 提供)。
  • OpenVINO 支持:YOLOv5 ONNX 模型现已兼容。借助 OpenVINO,模型现在可以充分利用 Intel CPU 和集成 GPU 的强大功能,适用于各种应用场景 (#6057,由 @glenn-jocher 提供)。
  • 导出基准测试:我们引入了一个新的基准测试工具,通过 python utils/benchmarks.py --weights yolov5s.pt 来评估所有 YOLOv5 导出格式的 mAP (Mean Average Precision) 和速度。目前该工具在 CPU 上运行,我们计划在未来的更新中增加 GPU 基准测试 (#6613,由 @glenn-jocher 提供)。
  • 超参数:我们的超参数进行了一个微小但关键的调整——在 hyp-scratch-large.yaml 中,学习率因子 (lrf) 已从 0.2 降低到 0.1 (#6525,由 @glenn-jocher 提供)。
  • 训练:默认的学习率 (LR) 调度程序已更新为 one-cycle 线性,取代了之前基于余弦退火的 one-cycle,以获得更好的训练结果 (#6729,由 @glenn-jocher 提供)。

YOLOv5 v6.1 features

为了展示我们对不同格式的全方位支持,YOLOv5 现在正式支持 11 种格式。这不仅支持导出,还支持通过 detect.py 和 PyTorch Hub 进行推理,以及通过验证来评估 mAP 和速度:

  • ✅ PyTorch
  • ✅ TorchScript
  • ✅ ONNX
  • ✅ OpenVINO
  • ✅ TensorRT
  • ✅ CoreML
  • ✅ TensorFlow SavedModel
  • ✅ TensorFlow GraphDef
  • ✅ TensorFlow Lite
  • ✅ TensorFlow Edge TPU
  • ✅ TensorFlow.js

Link to this section共同为大众打造 AI#

在 Ultralytics,驱动我们的不仅仅是领导地位,更是参与并回馈社区的热情。YOLOv5 系列在我们的旅程中发挥了重要作用,陪伴我们走过了胜利与挑战。这次更新是一项集体成就,代表了 来自 48 位新贡献者的 271 个 PR 的辛勤工作。我们坚定不移地致力于实现 AI 平民化,使其对每个人都易于访问和操作。

Link to this section加入 Vision AI 革命#

我们一直在寻找人才加入我们的行列,并欢迎各界参与我们的开源项目。如果你有兴趣成为最具突破性的 AI 团队的一员,请浏览我们的 职业页面 或考虑 为 YOLOv5 做出贡献

Link to this section从 AI 爱好者到 2022 年最受欢迎的目标检测#

今年,我们的 Ultralytics/YOLOv5 存储库在 GitHub 星标总数上超越了 Joseph Redmon 的 pjreddie/darknet YOLOv3,取得了里程碑式的进展,目前已超过 2.24 万星标。这证明了社区的信任和热情,也激励我们不断突破 Vision AI 的界限。我们深感荣幸能够传承 You Only Look Once 的愿景。

请访问我们的 YOLOv5 GitHub 存储库 获取关于新版本的详细信息,并加入充满活力的 YOLO 目标检测爱好者社区。

Link to this section体验零代码 YOLO 的魔力#

还有更多内容!如果你是计算机视觉领域的新手,或者更喜欢零代码体验,Ultralytics HUB 是你的入口。了解如何只需点击几下即可利用 YOLO 和计算机视觉技术。欲了解更多信息,请访问 Ultralytics HUB - 通往 AI 的门户,开启你的计算机视觉之旅。

Explore solutions

Real-time AI that works with your team

机器人技术中的 AI

使用 Ultralytics YOLO 模型为智能机器赋能。机器人技术中的视觉 AI 可推动自主导航、感知、物体跟踪和实时控制。

了解更多
Real-time AI that works with your team

物流中的 AI

使用 Ultralytics YOLO 模型简化物流。视觉 AI 可实现包裹检查、分拣、车辆跟踪和实时仓库安全监控。

了解更多
Real-time AI that works with your team

零售业 AI

使用 Ultralytics YOLO 模型重塑零售业。视觉 AI 推动库存跟踪、货架监控、队列管理和更智能的客户洞察。

了解更多
Real-time AI that works with your team

医疗保健中的 AI

利用 Ultralytics YOLO 模型构建医疗保健解决方案。医疗保健中的视觉 AI 可助力更快速的医学影像分析、更智能的诊断和患者监测。

了解更多
Real-time AI that works with your team

制造业中的 AI

使用 Ultralytics YOLO 模型优化制造业。视觉 AI 推动质量控制、缺陷检测、PPE 合规性和装配线自动化。

了解更多
Real-time AI that works with your operation

汽车中的 AI

将计算机视觉应用于汽车行业,并配合 Ultralytics YOLO 模型。汽车视觉 AI 可提升道路安全、辅助驾驶和车辆自动化,打造更智能的道路。

了解更多
Real-time AI tailored to your operation

农业中的 AI

借助 Ultralytics YOLO 模型,将视觉 AI 引入智慧农业。赋能作物监测、牲畜追踪和精准农业,实现更高、更智能的产量。

了解更多
Real-time AI that works with your team

机器人技术中的 AI

使用 Ultralytics YOLO 模型为智能机器赋能。机器人技术中的视觉 AI 可推动自主导航、感知、物体跟踪和实时控制。

了解更多
Real-time AI that works with your team

物流中的 AI

使用 Ultralytics YOLO 模型简化物流。视觉 AI 可实现包裹检查、分拣、车辆跟踪和实时仓库安全监控。

了解更多
Real-time AI that works with your team

零售业 AI

使用 Ultralytics YOLO 模型重塑零售业。视觉 AI 推动库存跟踪、货架监控、队列管理和更智能的客户洞察。

了解更多
Real-time AI that works with your team

医疗保健中的 AI

利用 Ultralytics YOLO 模型构建医疗保健解决方案。医疗保健中的视觉 AI 可助力更快速的医学影像分析、更智能的诊断和患者监测。

了解更多
Real-time AI that works with your team

制造业中的 AI

使用 Ultralytics YOLO 模型优化制造业。视觉 AI 推动质量控制、缺陷检测、PPE 合规性和装配线自动化。

了解更多
Real-time AI that works with your operation

汽车中的 AI

将计算机视觉应用于汽车行业,并配合 Ultralytics YOLO 模型。汽车视觉 AI 可提升道路安全、辅助驾驶和车辆自动化,打造更智能的道路。

了解更多
Real-time AI tailored to your operation

农业中的 AI

借助 Ultralytics YOLO 模型,将视觉 AI 引入智慧农业。赋能作物监测、牲畜追踪和精准农业,实现更高、更智能的产量。

了解更多
Real-time AI that works with your team

机器人技术中的 AI

使用 Ultralytics YOLO 模型为智能机器赋能。机器人技术中的视觉 AI 可推动自主导航、感知、物体跟踪和实时控制。

了解更多
Real-time AI that works with your team

物流中的 AI

使用 Ultralytics YOLO 模型简化物流。视觉 AI 可实现包裹检查、分拣、车辆跟踪和实时仓库安全监控。

了解更多
Real-time AI that works with your team

零售业 AI

使用 Ultralytics YOLO 模型重塑零售业。视觉 AI 推动库存跟踪、货架监控、队列管理和更智能的客户洞察。

了解更多
Real-time AI that works with your team

医疗保健中的 AI

利用 Ultralytics YOLO 模型构建医疗保健解决方案。医疗保健中的视觉 AI 可助力更快速的医学影像分析、更智能的诊断和患者监测。

了解更多
Real-time AI that works with your team

制造业中的 AI

使用 Ultralytics YOLO 模型优化制造业。视觉 AI 推动质量控制、缺陷检测、PPE 合规性和装配线自动化。

了解更多
Real-time AI that works with your operation

汽车中的 AI

将计算机视觉应用于汽车行业,并配合 Ultralytics YOLO 模型。汽车视觉 AI 可提升道路安全、辅助驾驶和车辆自动化,打造更智能的道路。

了解更多
Real-time AI tailored to your operation

农业中的 AI

借助 Ultralytics YOLO 模型,将视觉 AI 引入智慧农业。赋能作物监测、牲畜追踪和精准农业,实现更高、更智能的产量。

了解更多

让我们一起构建 AI 的未来!

开启你的机器学习未来之旅