Ultralytics YOLO27:
Get Started

借助 YOLO26 与 TensorBoard 的集成获取视觉洞察#

深入了解计算机视觉模型(例如 Ultralytics 的 YOLO26)并对其进行微调时,仔细查看训练过程会让一切变得更加简单。模型训练可视化有助于你了解模型的学习模式、性能指标和整体行为。YOLO26 与 TensorBoard 的集成让可视化和分析过程更加轻松,也使你能够更高效、更有依据地调整模型。

本指南介绍如何将 TensorBoard 与 YOLO26 配合使用。你将了解多种可视化方式,从跟踪指标到分析模型计算图。这些工具可以帮助你更好地了解 YOLO26 模型的性能。

TensorBoard#

TensorBoard 是 TensorFlow 的可视化工具包,是开展机器学习实验的重要工具。TensorBoard 提供了一系列可视化工具,对于监控机器学习模型至关重要。这些工具包括跟踪损失和精度等关键指标、可视化模型计算图,以及查看权重和偏置随时间变化的直方图。它还支持将嵌入向量投影到低维空间,并显示多媒体数据。

使用 TensorBoard 训练 YOLO26#

使用 TensorBoard 训练 YOLO26 模型十分简单,而且优势显著。

安装#

要安装所需软件包,请运行:

安装
# Install YOLO26 and TensorBoard
pip install ultralytics tensorboard

TensorBoard 是可选依赖项。安装后,使用 yolo settings tensorboard=True 启用一次回调;此设置会在后续训练运行中持续生效。

有关安装过程的详细说明和最佳实践,请务必查看我们的 YOLO26 安装指南。安装 YOLO26 所需的软件包时,如果遇到任何问题,请参阅我们的常见问题指南,获取解决方案和建议。

为 Google Colab 配置 TensorBoard#

使用 Google Colab 时,请务必在运行训练代码前配置 TensorBoard:

为 Google Colab 配置 TensorBoard
%load_ext tensorboard
%tensorboard --logdir path/to/runs

用法#

开始阅读使用说明前,请务必了解 Ultralytics 提供的 YOLO26 模型,以便根据项目需求选择最合适的模型。

启用或停用 TensorBoard

默认情况下,TensorBoard 日志记录处于停用状态。你可以使用 yolo settings 命令启用或停用日志记录。

# Enable TensorBoard logging
yolo settings tensorboard=True

# Disable TensorBoard logging
yolo settings tensorboard=False
用法
from ultralytics import YOLO

# 加载预训练模型
model = YOLO("yolo26n.pt")

# 训练模型
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)

运行上面的用法代码片段后,你会看到以下输出:

TensorBoard: Start with 'tensorboard --logdir runs/detect/train', view at http://localhost:6006/

此输出表明 TensorBoard 正在监控你的 YOLO26 训练过程。你可以在 localhost:6006 访问 TensorBoard 信息面板,查看实时训练指标和模型性能。对于使用 Google Colab 的用户,TensorBoard 会显示在你运行 TensorBoard 配置命令的同一单元格中。

如需了解更多模型训练过程的相关信息,请务必查看我们的 YOLO26 模型训练指南。如果你想进一步了解日志记录、检查点、绘图和文件管理,请阅读我们的配置使用指南。

了解 YOLO26 训练的 TensorBoard#

现在,我们来了解 TensorBoard 在 YOLO26 训练中的各种功能和组件。TensorBoard 的三个主要部分是时间序列、标量和计算图。

时间序列#

TensorBoard 的时间序列功能可以动态、细致地展示 YOLO26 模型各项训练指标随时间的变化。它着重呈现各项指标在不同训练轮次中的变化过程和趋势。以下是你可能看到的示例。

TensorBoard 时间序列训练指标可视化

TensorBoard 时间序列的主要功能#

  • 筛选标签和固定卡片:此功能允许用户筛选特定指标,并固定卡片以便快速比较和查看。它对于关注训练过程的特定方面尤其有用。

  • 详细指标卡片:时间序列会将指标划分为不同类别,例如学习率(lr)、训练(train)和验证(val)指标,并为每项指标显示单独的卡片。

  • 图形化展示:时间序列部分中的每张卡片都会显示某个指标在整个训练过程中的详细图表。此可视化表示有助于发现训练过程中的趋势、模式或异常。

  • 深入分析:时间序列可对每项指标进行深入分析。例如,它会显示不同的学习率区间,让你了解学习率调整会如何影响模型的学习曲线。

时间序列在 YOLO26 训练中的重要性#

时间序列部分对于全面分析 YOLO26 模型的训练进度至关重要。它可以让你实时跟踪指标,及时发现并解决问题。它还会详细展示每项指标的变化过程,这对于微调模型和提升模型性能至关重要。

标量#

TensorBoard 中的标量对于绘制和分析 YOLO26 模型训练过程中的损失和精度等简单指标至关重要。标量可以清晰、简洁地展示这些指标在每个训练轮次中的变化,让你了解模型学习的有效性和稳定性。以下是你可能看到的示例。

显示 YOLO 训练指标的 TensorBoard 标量仪表板

TensorBoard 标量的主要功能#

  • 学习率 (lr) 标签:这些标签显示不同阶段(例如 lr/pg0、lr/pg1、lr/pg2)的学习率变化情况。这有助于我们了解学习率调整对训练过程的影响。

  • 指标标签:标量包括以下性能指标:

    • metrics/mAP50(B):50% 交并比 (IoU) 下的平均精确率,这是评估目标检测准确率的关键指标。

    • metrics/mAP50-95(B):平均精确率,在一系列 IoU 阈值下计算,可更全面地评估准确率。

    • metrics/precision(B):表示正确预测的正例观测值所占的比例,是了解预测准确率的关键指标。

    • metrics/recall(B):对于漏检影响重大的模型,这项指标很重要,用于衡量模型检测所有相关实例的能力。

    • 如需了解不同指标的更多信息,请阅读我们的性能指标指南。

  • 训练和验证标签(train、val):这些标签专门显示训练集和验证集的指标,便于对比模型在不同数据集上的表现。

监控标量的重要性#

观察标量指标对于微调 YOLO26 模型至关重要。这些指标的变化(例如损失曲线中的尖峰或不规则模式)可能揭示潜在问题,例如过拟合、欠拟合或学习率设置不当。通过密切监控这些标量,你可以做出明智的决策来优化训练过程,确保模型有效学习并达到预期性能。

标量与时间序列的区别#

TensorBoard 中的标量和时间序列都用于跟踪指标,但用途略有不同。标量侧重于将损失和准确率等简单指标绘制为标量值,概览这些指标在每个训练轮次中的变化情况。TensorBoard 的时间序列部分则提供各种指标的详细时间线视图。它尤其适用于监控指标随时间推移的进展和趋势,让你深入了解训练过程的具体情况。

计算图#

TensorBoard 的计算图部分会将 YOLO26 模型的计算图可视化,展示模型内部的运算和数据流。这是一个强大的工具,可用于了解模型结构、确保所有层正确连接,以及识别数据流中可能存在的瓶颈。以下是你可能看到的示例。

YOLO 模型的 TensorBoard 计算图可视化

计算图对于调试模型尤其有用,特别是调试 YOLO26 等深度学习模型中常见的复杂架构时。计算图有助于验证层之间的连接和模型的整体设计。

总结#

本指南旨在帮助你使用 TensorBoard 可视化和分析机器学习模型训练情况。指南重点介绍 TensorBoard 的关键功能如何帮助你了解 YOLO26 训练过程中的训练指标和模型性能。

如需深入了解这些功能及其有效使用策略,请参阅 TensorFlow 官方的 TensorBoard 文档和 GitHub 仓库。

想进一步了解 Ultralytics 的各种集成方案吗?查看 Ultralytics 集成指南页面,了解还有哪些值得探索的强大功能!

常见问题#

  • 将 TensorBoard 与 YOLO26 搭配使用,可以获得多种对高效训练模型至关重要的可视化工具:

    • 实时指标跟踪:实时跟踪损失、准确率、精确率和召回率等关键指标。
    • 模型计算图可视化:通过可视化计算图来了解和调试模型架构。
    • 嵌入可视化:将嵌入投影到低维空间,以便更好地理解数据。

    借助这些工具,你可以有针对性地调整模型,从而提升 YOLO26 的性能。如需详细了解 TensorBoard 的功能,请查看 TensorFlow 的 TensorBoard 指南。

  • 使用 TensorBoard 训练 YOLO26 模型并监控训练指标,请按以下步骤操作:

    1. 安装 TensorBoard 和 YOLO26:运行 pip install ultralytics tensorboard。
    2. 启用 TensorBoard 日志记录:运行 yolo settings tensorboard=True。训练期间,YOLO26 会将指标写入运行目录。
    3. 启动 TensorBoard:运行 tensorboard --logdir path/to/runs 启动 TensorBoard,并将其指向训练运行目录(例如 runs/detect/train)。

    访问 localhost:6006 即可打开 TensorBoard 仪表板,实时查看各种训练指标。如需深入了解训练配置,请访问我们的 YOLO26 配置指南。

  • 训练 YOLO26 模型时,你可以使用 TensorBoard 可视化一系列重要指标,包括:

    • 损失(训练和验证):表示模型在训练和验证期间的表现。
    • 准确率/精确率/召回率:用于评估检测准确率的关键性能指标。
    • 学习率:跟踪学习率变化,了解其对训练动态的影响。
    • mAP(平均精确率):用于全面评估不同 IoU 阈值下的目标检测准确率。

    这些可视化对于跟踪模型性能并进行必要的优化至关重要。如需了解这些指标的更多信息,请参阅我们的性能指标指南。

  • 可以,你可以在 Google Colab 环境中使用 TensorBoard 训练 YOLO26 模型。以下是快速设置方法:

    为 Google Colab 配置 TensorBoard
    %load_ext tensorboard
    %tensorboard --logdir path/to/runs

    然后,运行 YOLO26 训练脚本:

    from ultralytics import YOLO
    
    # 加载预训练模型
    model = YOLO("yolo26n.pt")
    
    # 训练模型
    results = model.train(data="coco8.yaml", epochs=100, imgsz=640)

    TensorBoard 会在 Colab 中可视化训练进度,让你实时查看损失和准确率等指标。如需了解配置 YOLO26 训练的更多详情,请参阅我们的 YOLO26 安装指南。

评论