深圳Yolo 视觉
深圳
立即加入

使用Ultralytics YOLO11 和 DVC 进行实验跟踪

Abirami Vina

5 分钟阅读

2025年2月25日

了解如何使用实验跟踪来简化与 DVC 集成的Ultralytics YOLO11 实验,从而提高模型性能。

跟踪和监控计算机视觉实验(训练机器解释和理解视觉数据)是开发和微调视觉人工智能模型的关键部分,例如 Ultralytics YOLO11.这些实验通常需要测试不同的关键参数,并记录多个模型训练运行的指标和结果。这样做有助于分析模型性能,并根据数据对模型进行改进。 

如果没有定义完善的实验跟踪系统,比较结果和更改模型可能会变得复杂并导致错误。事实上,自动化此过程是一个不错的选择,可以确保更好的一致性。

这正是Ultralytics 支持的DVCLive 集成所要实现的目标。DVCLive 提供了一种简化的方式,可在单一工作流程中自动记录实验细节、可视化结果并管理模型性能跟踪。

在本文中,我们将讨论如何在训练Ultralytics YOLO11 时使用 DVCLive 集成。我们还将了解它的优势,以及它如何使实验跟踪更容易,从而更好地开发视觉人工智能模型

什么是DVCLive?

由 DVC(数据版本控制)创建的 DVCLive 是一款可靠的开源工具,专为跟踪机器学习实验而设计。DVCLivePython 库提供了一个实时实验记录器,使人工智能开发人员和研究人员track 其实验的指标和参数成为可能。 

例如,它可以自动记录关键的模型性能指标,比较跨训练运行的结果,并可视化模型性能。这些功能使DVCLive能够帮助您维护结构化且可重复的机器学习工作流程。

图 1. 快速浏览 DVCLive 的实验跟踪仪表板。

DVCLive 的主要特性

DVCLive 集成易于使用,通过提供清晰易懂的数据可视化和分析工具,可以改进您的计算机视觉项目。以下是 DVCLive 的其他一些主要功能:

  • 支持各种框架: DVCLive 可以与其他流行的机器学习框架一起使用。这使得将其整合到现有工作流程中并提高实验跟踪能力变得简单。
  • 交互式绘图:它可以用于从数据自动生成交互式绘图,从而提供性能指标随时间变化的直观表示。 
  • 轻量级设计:DVCLive 是一个轻量级、灵活且易于访问的库,因为它可以在不同的项目和环境中使用。

为什么要使用 DVCLive 集成?

当您阅读Ultralytics 文档并探索可用的集成时,您可能会问自己:DVCLive集成与众不同的地方是什么?是什么让 DVCLive 集成与众不同,为什么我的工作流程要选择它?

TensorBoardMLflow等集成也提供了用于跟踪指标和可视化结果的工具,因此了解使这种集成脱颖而出的独特品质至关重要。 

以下是 DVCLive 成为Ultralytics YOLO 项目理想选择的原因:

  • 最小开销: DVCLive 是一个用于记录实验指标的绝佳工具,无需添加任何额外的计算或存储负载。它将日志保存为纯文本或 JSON 文件,使其易于集成到现有工作流程中,而无需依赖外部服务或数据库。
  • 与 DVC 原生集成:DVCLive 由 DVC 背后的团队构建,可与 DVC 的数据和模型版本系统顺利协作。它还允许用户track 数据集版本、模型检查点和管道更改,是已经使用 DVC 实现机器学习可重复性的团队的理想选择。
  • 与 Git 兼容:DVCLive 与 Git 集成,可轻松track 更改、比较模型和恢复到以前的版本,同时保持实验数据的有序性和版本控制。

DVCLive 入门 

使用 DVCLive 跟踪Ultralytics YOLO11 模型训练比您想象的要简单得多。一旦安装和配置了必要的库,您就可以快速开始自定义训练YOLO11 模型。

训练后,您可以调整关键设置,如epochs(模型遍历整个数据集的次数)、patience(如果没有改进,等待停止的时间)和目标图像大小(用于训练的图像的分辨率)以提高准确性。然后,您可以使用DVCLive的可视化工具来比较不同版本的模型并分析它们的性能。

如需更详细地了解模型培训流程和最佳实践,请查看我们的Ultralytics YOLO 模型定制培训文档。

接下来,让我们一起来了解如何在定制培训YOLO11 时安装和使用 DVCLive 集成。

安装所需组件

在开始培训YOLO11 之前,您需要安装Ultralytics Python 软件包和 DVCLive。这种集成的设计方式使两个库在默认情况下无缝协作,因此您无需担心复杂的配置。

如下图所示,只需一条 Pip 命令就能在几分钟内完成整个安装过程,Pip 是一种用于安装Python 库的软件包管理工具。 

图 2.安装Ultralytics 和 DVCLive。

安装完软件包后,您就可以设置环境并添加必要的凭证,以确保 DVCLive 顺利运行。设置一个 Git 仓库也有助于track 代码和对 DVCLive 设置的任何更改。 

有关详细的分步说明和其他有用的提示,请查看我们的安装指南。如果您在安装所需软件包时遇到任何问题,我们的常见问题指南提供了解决方案和资源来帮助您。

使用 DVCLive 实验训练 

完成YOLO11 模型训练后,您可以使用可视化工具深入分析结果。具体来说,您可以使用 DVC 的应用程序接口来提取数据,并用Pandas (Python 库,使数据处理更容易,如将数据整理成表格以便分析和比较)进行处理,以便更轻松地进行处理和可视化。 

为了以更具互动性和可视化的方式探索您的结果,您还可以尝试使用 Plotly 的平行坐标图(一种显示不同模型参数和性能结果之间如何关联的图表)。 

最终,您可以利用这些可视化的洞察力,在优化模型、调整超参数或进行其他修改以提升整体性能方面做出更好的决策。 

YOLO11 和 DVCLive 集成的应用

既然我们已经学会了如何使用 DVCLive 集成安装和可视化YOLO11 培训结果,那就让我们来探索一下该集成可以增强的一些应用。

农业和精准农业

当涉及到农业和收获粮食作物时,精确度可以发挥巨大作用。例如,农民可以利用YOLO11对物体检测和实例分割的支持来识别潜在的作物疾病、track 牲畜和detect 虫害。 

特别是,YOLO11 可以通过分析无人机或摄像机拍摄的图像,帮助捕捉植物病害、有害害虫或不健康动物的早期迹象。这类视觉人工智能系统能让农民迅速采取行动,阻止问题蔓延,从而节省时间并减少损失。

图 3.使用YOLO11 监测农作物的示例。

由于农场条件会随着天气和季节的变化而不断变化,因此在各种图像上测试模型以确保其在不同情况下的良好表现非常重要。使用 DVCLive 集成为农业应用定制培训YOLO11 是track 其性能的好方法,尤其是在使用各种数据集时。 

零售业客户行为分析

零售商店可以利用 AI 和计算机视觉来了解客户行为,并进行改进以提升购物体验。 

通过分析监控摄像头的视频,YOLO11 可以track 人们如何在店内走动、哪些区域人流量最大以及购物者如何与产品互动。然后,这些数据可用于制作热图,展示哪个货架最吸引人、顾客在不同过道上花费的时间以及广告展示是否受到关注。 

借助这种商业智能,店主可以重新布置产品以增加销售额,加快结账队伍的速度,并调整人员配置以在最需要的时间和地点为顾客提供帮助。

图 4.使用YOLO11 绘制的购物中心热图示例。

通常情况下,零售店具有独特的特点,例如不同的照明条件、布局和人群规模。由于这些差异,用于分析商店活动的计算机视觉模型需要针对每个地点进行仔细的测试和调整,以确保准确性。例如,DVCLive 集成可以帮助微调YOLO11,使其在零售应用中更加精确和可靠,从而能够更好地洞察客户行为和商店运营。

主要要点

在使用 DVCLive 集成的同时对YOLO11 进行自定义训练,可以更轻松地track 和改进计算机视觉实验。它会自动记录重要细节,显示清晰的视觉结果,并帮助您比较模型的不同版本。 

无论您是想提高农场生产力,还是改善商店的购物体验,此集成都能确保您的 Vision AI 模型表现良好。通过实验跟踪,您可以系统地测试、改进和优化您的模型,从而不断提高准确性和性能。

加入我们的社区,并探索我们的GitHub仓库,以了解更多关于视觉人工智能的信息,并查看我们的许可选项,以启动您的计算机视觉项目。对人工智能在制造业中的应用计算机视觉在自动驾驶中的应用等创新感兴趣吗?访问我们的解决方案页面以了解更多信息。 

让我们一起构建人工智能的未来!

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

免费开始