Ultralytics YOLO27: 加入候补名单。
获取 YOLO 许可
集成

使用计算机视觉数据集对 Ultralytics YOLO11 进行自定义训练

了解 Roboflow 集成如何让开源计算机视觉数据集更易获取,从而简化 Ultralytics YOLO11 的自定义训练。

Abirami VinaAbirami Vina6 分钟阅读
了解 Roboflow 集成如何让开源计算机视觉数据集更易获取,从而简化 Ultralytics YOLO11 的自定义训练。

训练 Ultralytics YOLO11 这类计算机视觉模型,通常需要收集数据集图像、标注图像、准备数据,并微调模型以满足你的具体需求。Ultralytics Python 包让这些步骤简单易用,但视觉 AI 开发仍可能耗费大量时间。

当你需要赶紧完成任务或开发原型时,这一点尤为明显。在这种情况下,能够简化部分流程的工具或集成(例如简化数据集准备或自动执行重复性任务)可以带来很大帮助。通过减少所需的时间和精力,这些解决方案能让你专注于构建和改进模型。这正是 Roboflow 集成所能提供的。

Roboflow 集成让你可以轻松访问 Roboflow Universe 中的数据集。Roboflow Universe 是一个大型开源计算机视觉数据集库。你不必花几个小时收集和整理数据,只需快速查找并使用现有数据集,就能启动 Ultralytics YOLO11 训练流程。这项集成让你能更快、更轻松地试验和迭代计算机视觉模型开发。

本文将深入介绍如何利用 Roboflow 集成,加快模型开发。

什么是 Roboflow Universe?#

Roboflow Universe 是由 Roboflow 维护的平台。Roboflow 是一家致力于简化计算机视觉开发的公司。该平台包含超过 3.5 亿张图像、50 万个数据集和 10 万个微调模型,支持目标检测、分割和图像分类等任务。Roboflow Universe 汇集了全球开发人员和研究人员的贡献,是任何希望快速启动或改进计算机视觉项目的人的协作中心。

Roboflow Universe 上的目标检测数据集示例

图 1。Roboflow Universe 上的目标检测数据集示例。

Roboflow Universe 包含以下主要功能:

  • 数据集探索工具:探索、筛选和可视化数据集,快速找到符合项目要求的资源。
  • 导出选项:将数据导出为 COCO、YOLO、TFRecord、CSV 等格式,以适配你的工作流。
  • 数据集分析:使用分析工具深入了解各种数据集,查看标签分布、类别不平衡和数据集质量等可视化信息。
  • 版本跟踪:查看并访问贡献者上传的数据集的不同版本,从而跟踪更新、比较更改,并选择最符合项目需求的版本。

Roboflow 集成帮助你找到合适的数据#

找到合适的数据集通常是构建计算机视觉模型最具挑战性的环节之一。创建数据集通常需要收集大量图像、确保图像与任务相关,然后准确标注图像。

这个过程可能耗费大量时间和资源,尤其是你需要在短时间内尝试不同方案时。即使是寻找现有数据集也可能很棘手,因为这些数据集往往分散在不同平台、缺少完善的文档,或没有你需要的特定标注。

例如,如果你正在构建一个用于检测农田杂草的计算机视觉应用,可能想测试不同的视觉 AI 方法,例如目标检测和实例分割。这样,你就能先试验并找出最合适的方法,再投入时间和精力收集、标注自己的数据集。

使用 Ultralytics YOLO11 检测汽车零件

图 2。使用 Ultralytics YOLO11 检测汽车零件。

使用 Roboflow 集成,你可以浏览各种农业相关数据集,包括专注于杂草检测、作物健康或田间监测的数据集。这些现成的数据集让你无需预先投入精力创建自己的数据,就能尝试不同技术并改进模型。

Roboflow 集成的工作方式#

介绍完如何使用 Roboflow 集成查找合适的数据集后,我们来看看它如何融入你的工作流。从 Roboflow Universe 中选定数据集后,你可以将其导出或下载为 YOLO11 格式。导出数据集后,你就可以使用 Ultralytics Python 包对 YOLO11 进行定制训练。

下载数据集时,你可能会注意到 Roboflow Universe 也支持用于训练其他模型的格式。那么,为什么要选择定制训练 Ultralytics YOLO11?

本文撰写时,YOLO11 是 Ultralytics YOLO 模型的最新版本(此后已被 Ultralytics YOLO26 取代),旨在实现更快、更准确的目标检测。YOLO11m 的参数量(模型在训练期间调整以进行预测的内部数值)比 YOLOv8m 少 22%,但在 COCO 数据集上的平均精度均值(mAP)更高。速度与精度之间的这种平衡,让 YOLO11 成为众多计算机视觉应用的灵活选择,尤其适合针对特定任务进行自定义模型训练。

以下是 定制训练 Ultralytics YOLO11 的具体流程:

  • 输入数据:Ultralytics YOLO11 模型会处理你的数据集,并从图像及其标注中学习检测和分类物体。
  • 预测和反馈:模型会预测图像中的物体,并将预测结果与数据集中提供的正确答案进行比较。
  • 性能跟踪:系统会监控精确率(正确检测的比例)、召回率(找到的实际物体比例)和损失(预测误差)等指标,以评估进展。
  • 迭代学习:模型会在多个轮次(epoch)中调整参数,以提高检测准确率并尽量减少错误。
  • 最终模型输出:训练完成后,经过优化的模型会保存下来,可随时部署。

专注于计算机视觉开发的其他集成#

探索 Roboflow 集成时,你会注意到 Ultralytics 文档中还介绍了其他集成。我们支持与计算机视觉开发不同阶段相关的多种集成。

这是为了给社区提供多种选择,让你可以根据具体工作流选择最适合的方案。

Ultralytics 支持的集成概览

图 3。 Ultralytics 支持的集成概览。

除了数据集之外,Ultralytics 支持的其他集成也覆盖计算机视觉流程的不同环节,例如训练、部署和优化。以下是我们支持的其他集成示例:

  • 训练集成:Amazon SageMaker 和 Paperspace Gradient 等集成提供基于云的平台,可简化训练工作流,让模型开发和测试更加高效。
  • 工作流和实验跟踪集成:ClearML、MLflow 和 Weights & Biases(W&B)有助于自动化工作流、跟踪实验并改进协作,让机器学习项目更易于管理。
  • 优化和部署集成:CoreML、ONNX 和 OpenVINO 支持在各种设备和框架上进行优化部署,确保模型在 Apple 硬件和 Intel CPU 等平台上高效运行。
  • 监控和可视化集成:TensorBoard 和 Weights & Biases 提供可视化训练进度和监控性能的工具,让你深入了解模型,从而对其进行改进。

Ultralytics YOLO11 的应用及集成的作用#

支持计算机视觉开发的集成方案,加上 Ultralytics YOLO11 可靠的能力,让解决现实挑战变得更加轻松。不妨看看制造业中的计算机视觉等创新应用:视觉 AI 可用于检测生产线上的缺陷,例如金属零件上的划痕或缺失的组件。收集这类任务所需的数据往往耗时且困难,因为通常需要进入特定的专业环境。

通常需要在生产线上安装摄像头或传感器,以采集产品图像。为了确保图像清晰且一致,需要大量采集图像,并且通常要保持光照和拍摄角度一致。

采集图像后,必须仔细标注,为每种缺陷(例如划痕、凹痕或缺失组件)添加准确标签。这个过程需要投入大量时间和资源,也需要专业知识,才能确保数据集准确反映现实世界的多样性。创建稳健可靠的数据集时,必须考虑缺陷尺寸、形状和材料等因素的差异。

提供现成数据集的集成方案有助于完成工业质量控制等任务。结合 Ultralytics YOLO11 的实时检测能力,制造商可以监控生产线、即时发现缺陷并提高效率。

使用 Ultralytics YOLO11 检测并计数正在生产的罐头

图 4. 使用 Ultralytics YOLO11 检测并计数正在生产的罐头的示例。

除了制造业,数据集相关的集成方案也可应用于许多其他行业。企业将 Ultralytics YOLO11 的速度和准确性与易于获取的数据集结合起来,就能快速开发并部署满足特定需求的解决方案。例如,在医疗保健领域,数据集集成有助于开发分析医学图像、检测肿瘤等异常情况的解决方案。同样,在自动驾驶领域,这类集成方案有助于识别车辆、行人和交通标志,从而提升安全性。

要点#

寻找合适的数据集往往是构建计算机视觉模型时最耗时的环节之一。不过,Roboflow 集成让你更容易为 Ultralytics YOLO 模型定制训练找到最佳数据集,即使你刚接触计算机视觉也不例外。

Roboflow Universe 汇集了大量计算机视觉任务数据集,包括目标检测、图像分类和实例分割数据集,让数据发现过程变得轻松省事。它能帮助你快速上手,将精力集中在构建模型上,而不是花时间收集和整理数据。这种精简流程能帮助开发人员更高效地构建原型、迭代并开发计算机视觉解决方案。

如需了解更多信息,请访问我们的 GitHub 代码库并参与社区交流。你还可以在我们的解决方案页面上,探索自动驾驶汽车中的 AI和农业中的计算机视觉等领域的创新。

作者

Abirami Vina
Abirami Vina

内容创作者 · Ultralytics

Abirami Vina 职业生涯始于计算机视觉工程师,由此在视觉 AI 和机器学习领域打下了坚实基础。她是 Ultralytics 的内容创作者,也是 Scribe of AI 的创始人兼首席撰稿人,为专注 AI 的公司撰写高质量技术内容。她热衷于让 AI 更易理解、更具吸引力,通过自己的作品连接前沿技术与现实影响。

评论

提出问题、分享你的结果,或帮助其他人。

探索解决方案

体育领域的计算机视觉

体育领域的计算机视觉

了解 Ultralytics YOLO 如何通过球员和球追踪、表现分析、教练洞察及运动员康复,推动体育视觉技术的发展。
了解详情
航空影像计算机视觉

航空影像计算机视觉

使用 Ultralytics YOLO 将无人机和航空影像转化为农业、水产养殖、保护工作和地理空间分析的实时洞察。
了解详情
航空航天中的计算机视觉

航空航天中的计算机视觉

使用 Ultralytics YOLO 将计算机视觉应用于航空航天领域,包括飞机检查、跑道 FOD 检测、机场地面作业和复合材料缺陷检测。
了解详情
安防领域的计算机视觉

安防领域的计算机视觉

使用 Ultralytics YOLO 模型守护每个场所。视觉 AI 可用于周界监控、威胁检测、车牌识别和工作场所安全。
了解详情
机器人领域的计算机视觉

机器人领域的计算机视觉

使用 Ultralytics YOLO 模型提升智能机器的能力。机器人领域的视觉 AI 可用于自主导航、感知、目标跟踪和实时控制。
了解详情
物流中的计算机视觉

物流中的计算机视觉

借助 Ultralytics YOLO 模型优化物流流程。视觉 AI 可用于包裹检查、分拣、车辆跟踪和仓库安全实时监控。
了解详情
零售中的计算机视觉

零售中的计算机视觉

借助 Ultralytics YOLO 模型重新构想零售。视觉 AI 可用于库存跟踪、货架监控、排队管理和更深入的客户洞察。
了解详情
医疗领域的计算机视觉

医疗领域的计算机视觉

借助 Ultralytics YOLO 模型构建医疗解决方案。医疗领域的视觉 AI 可加快医学影像处理、优化诊断并监测患者。
了解详情
制造业中的计算机视觉

制造业中的计算机视觉

利用 Ultralytics YOLO 模型优化制造流程。视觉 AI 可用于质量控制、缺陷检测、个人防护装备合规检查和装配线自动化。
了解详情
汽车领域的计算机视觉

汽车领域的计算机视觉

利用 Ultralytics YOLO 模型在汽车领域应用计算机视觉。视觉 AI 可提升道路安全、驾驶辅助和车辆自动化水平,让道路更智能。
了解详情
农业中的计算机视觉

农业中的计算机视觉

借助 Ultralytics YOLO 模型,将视觉 AI 引入智慧农业。为作物监测、牲畜追踪和精准农业提供支持,从而提高产量并实现更智能的生产。
了解详情

让我们携手构建 AI 的未来!

开启你的旅程,迎接机器学习的未来