使用Weights & Biases优化Ultralytics YOLO模型训练

在众多计算机视觉应用中,Ultralytics YOLO11 已成为不可或缺的一部分。然而,训练、评估和部署这些复杂模型带来了许多挑战。跟踪关键训练指标、比较模型变体、分析模型行为和检测问题需要大量的工具和实验管理。

安装与配置

要开始使用 Weights & Biases 与Ultralytics YOLO11 的集成,首先需要安装必要的包。这可以通过以下命令完成:

pip install -U ultralytics wandb

对于安装过程中的详细指导和最佳实践,请参考 。如果在安装 YOLO11 所需的包时遇到任何困难,请查阅 以获得解决方案和提示。

设置 Weights & Biases

安装完必要的包后,下一步是设置Weights & Biases环境。这包括创建一个 Weights & Biases 账户并获取必要的 API 密钥,以便在开发环境和 W&B 平台之间建立顺畅的连接。

首先,在工作区初始化 Weights & Biases 环境。运行以下命令并按照消息提示进行初始化:

import wandb; wandb.login(key="<API_KEY>")

前往Weights & Biases认证页面生成并查找 API 密钥。使用此密钥通过 W&B 对环境进行认证。

使用 Weights & Biases 训练 YOLO11

在深入研究使用 Weights & Biases 训练 YOLO11 模型的使用说明之前,请务必查看 Ultralytics 提供的 。这将帮助为项目需求选择最合适的模型。

使用 Weights & Biases 可以为 YOLO11 训练过程带来效率和自动化。

from ultralytics import YOLO; model = YOLO("yolo11n.pt"); model.train(data="coco8.yaml", epochs=5, project="ultralytics", name="yolo11n")

以上代码将自动记录指标、超参数和模型工件到 W&B 项目中。

Weights & Biases仪表板

运行使用代码片段后,可以通过输出中提供的链接访问 Weights & Biases (W&B) 仪表板。此仪表板提供了模型训练过程的全面视图,使用 YOLO11。

W&B 仪表板的主要功能包括实时指标跟踪、超参数优化、比较分析、训练进度可视化、资源监控和模型工件管理。

沪ICP备2024098111号-1
上海秋旦网络科技中心:上海市奉贤区金大公路8218号1幢 联系电话:17898875485