敬请关注 YOLO Vision 2025!
2025年9月25日
英国夏令时 10:00 - 18:00
混合活动
Yolo Vision 2024

如何使用 Ultralytics HUB 训练您的自定义模型

Nuvola Ladi

3 分钟阅读

2024年5月31日

了解如何通过 Ultralytics HUB 轻松训练自定义 YOLO 模型。 这个循序渐进的指南将带您了解直观的平台,无缝集成、实时跟踪和云端训练。

在我们不断努力进行人工智能驱动的创新之际,请加入我们,一起仔细了解Ultralytics HUB,这是一个旨在简化训练自定义Ultralytics YOLO模型的平台。无论您是经验丰富的开发人员还是初学者,Ultralytics HUB 都能为您提供创建和管理计算机视觉模型的无缝体验。让我们一起探索如何在只需点击几下即可训练您的自定义 YOLO 模型。

什么是 Ultralytics HUB?

Ultralytics HUB 是一个综合平台,旨在处理您的数据集、项目和模型,从而更轻松地训练和部署机器学习模型。 

它与各种平台集成,支持通过其 移动应用程序(可在 Android 和 iOS 上使用)进行实时更新,并且随着新功能的不断发展。有关更多技术见解和逐步工作流程,请查看我们在 Medium 上发表的关于如何在云上训练计算机视觉模型的文章。

Ultralytics HUB 入门

用户友好的界面

您首先会注意到 Ultralytics HUB 具有直观的界面。主页可以轻松访问数据集、项目、模型集成。即使它仍处于 Beta 阶段,该平台已经拥有一系列强大的功能。

数据集和模型

Ultralytics HUB 预装了标准数据集,如 VOCCOCO 和 Simpsons,这些数据集通常用于基准测试。这些数据集非常庞大,其中 COCO 包含 80 个类别和 140,000 张图像。但是,该平台也支持较小的数据集,非常适合所有级别的用户。您可以上传您的自定义数据集或连接到外部工具(如 Roboflow)以进行图像标注和数据集准备。

创建项目

在 Ultralytics HUB 中创建一个新项目非常简单。以下是帮助您入门的分步指南:

  1. 创建项目:首先为您的项目命名。在此示例中,我们将其命名为“物体检测”。
  2. 选择数据集:从可用选项中选择一个数据集。为简单起见,我们将使用包含 14,000 张图片的 Simpsons 数据集。
  3. 选择模型:选择一个 YOLO 模型。Ultralytics HUB 支持各种模型,包括 YOLOv5YOLOv8。在本示例中,我们将选择 YOLOv8 nano 模型。
  4. 调整超参数:调整 epoch 数、图像大小和批量大小等设置。您还可以选择是使用 GPU 还是 CPU,并配置缓存选项。

训练模型

项目设置完成后,只需单击一下即可开始训练模型。Ultralytics HUB 提供多种训练选项:

  • 本地训练:在您的本地机器上安装 Ultralytics 并运行训练脚本。
  • Google Colab:打开一个预配置了必要代码的 Google Colab 笔记本。此选项无需编码,并在云端无缝运行。
  • HUB 云:Ultralytics HUB 云训练提供了一种无需代码即可训练 YOLO 模型的解决方案,非常适合非编码人员和企业主。 该工作流程包括数据集上传、模型选择和云实例设置,从而可以轻松地微调预训练模型并将其导出以用于各种应用。

在 Google Colab 中训练模型:

  1. 安装 Ultralytics:运行命令以在 notebook 中安装 Ultralytics。
  2. 设置和 API 密钥:配置设置并输入您的 API 密钥。
  3. 开始训练:执行训练命令,您的模型将开始训练。
__wf_reserved_inherit
图 1. Nicolai Nielsen 概述了如何在 Ultralytics HUB 上训练模型。

监控训练进度

Ultralytics HUB 提供模型训练进度的实时跟踪。您可以监控关键指标,如准确率、精确率、召回率和损失函数。该平台还可以可视化训练数据,让您了解模型如何随着时间的推移而改进。

高级功能和集成

云端训练和 RoboFlow 集成

Ultralytics HUB 引入了 云端训练,使用户可以直接在云端训练模型,而无需任何本地设置。该平台还与 Roboflow 集成,用于标注图像和准备数据集。这种集成简化了从数据准备到模型训练的工作流程。

移动应用

Ultralytics HUB 应用程序将实时对象检测带到您的移动设备。使用预训练模型,该应用程序可以高精度地检测 COCO 数据集中的对象,在 iPhone 14 Pro 上以每秒 30 帧的速度运行。此功能非常适合在实际场景中测试模型并展示您训练的模型的强大功能。

总结

Ultralytics HUB 改变了计算机视觉领域,使训练自定义 YOLO 模型比以往任何时候都更容易。其用户友好的界面、强大的功能和无缝集成使其成为开发人员和研究人员必备的工具。无论您是希望在本地、云端还是通过移动应用程序随时随地训练模型,Ultralytics HUB 都能满足您的需求。

还在等什么?立即进入 Ultralytics HUB,只需点击几下,即可释放您的机器学习项目的潜力!

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

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

免费开始
链接已复制到剪贴板