Ultralytics HUB 模型
Ultralytics HUB 模型为在自定义数据集上训练视觉AI模型提供了一个简化的解决方案。
该过程用户友好且高效,涉及一个简单的三步创建和由Ultralytics YOLOv8加速的训练。在训练过程中,可以实时获取模型指标的更新,以便您可以监控每个步骤的进展。一旦训练完成,您可以预览您的模型并轻松将其部署到实际应用中。因此,Ultralytics HUB 提供了一个全面且直接的系统,用于模型的创建、训练、评估和部署。
观看: Ultralytics HUB 训练和验证概述
训练模型
通过点击侧边栏中的 Models 按钮导航到 Models 页面,然后点击页面右上角的 Train Model 按钮。
Tip
您可以直接从 Home 页面训练模型。
此操作将触发 Train Model 对话框,该对话框包含三个简单步骤:
1. 数据集
在此步骤中,您需要选择要训练模型的数据集。选择数据集后,点击 Continue。
Tip
如果您直接从数据集页面训练模型,可以跳过此步骤。
2. 模型
在此步骤中,您需要选择要在其中创建模型的项目、模型的名称和模型的架构。
Note
Ultralytics HUB 将尝试预先选择项目。
如果您按照上述描述打开 Train Model 对话框,Ultralytics HUB 将预先选择您上次使用的项目。
如果您从项目页面打开 Train Model 对话框,Ultralytics HUB 将预先选择您所在的项目。
如果您还没有创建项目,您可以在此步骤中设置项目名称,它将与您的模型一起创建。
默认情况下,您的模型将使用预训练模型(在 COCO 数据集上训练)以减少训练时间。您可以通过打开 Advanced Model Configuration 手风琴来更改此行为并调整模型的配置。
Note
你可以轻松更改最常见的模型配置选项(例如训练轮数),但也可以使用自定义选项来访问与Ultralytics HUB相关的所有训练设置。
观看: 如何在Ultralytics HUB中配置Ultralytics YOLOv8训练参数
或者,你可以通过点击自定义标签,从之前训练过的模型开始训练。
当你对模型配置满意后,点击继续。
3. 训练
在这一步,你将开始训练你的模型。
Note
当你处于这一步时,你可以选择关闭训练模型对话框,稍后从模型页面开始训练你的模型。
Ultralytics HUB提供三种训练选项:
- Ultralytics云
- Google Colab
- 自带代理
a. Ultralytics云
你需要升级到专业计划以访问Ultralytics云。
要使用我们的云训练解决方案训练模型,请阅读Ultralytics云训练文档。
b. Google Colab
要使用Google Colab开始训练你的模型,请按照Ultralytics HUB训练模型对话框或Google Colab笔记本中显示的说明进行操作。
当训练开始时,你可以点击完成并在模型页面上监控训练进度。
Note
如果训练停止并且保存了检查点,你可以从模型页面恢复训练你的模型。
c. 自带代理
观看: 使用Ultralytics HUB进行自带代理模型训练
要使用你自己的代理开始训练你的模型,请按照Ultralytics HUB训练模型对话框中显示的说明进行操作。
从PyPI安装ultralytics
包。
接下来,使用提供的Python代码开始训练模型。
当训练开始时,你可以点击完成并在模型页面上监控训练进度。
Note
如果训练停止并且保存了检查点,您可以从模型页面恢复模型的训练。
分析模型
在您训练模型后,您可以分析模型指标。
训练标签展示了根据任务精心分组的最重要指标。
要访问所有模型指标,请点击图表标签。
Tip
每个图表都可以放大以获得更好的可视化效果。
此外,为了正确分析数据,您可以使用缩放功能。
预览模型
在您训练模型后,您可以通过点击预览标签来预览模型。
在测试卡片中,您可以从训练期间使用的数据集中选择预览图像,或从您的设备上传图像。
Note
您还可以使用相机直接拍摄照片并运行推理。
此外,您可以通过下载我们的Ultralytics HUB 应用,直接在您的iOS或Android移动设备上实时预览您的模型。
部署模型
在您训练模型后,您可以将其导出为13种不同的格式,包括ONNX、OpenVINO、CoreML、TensorFlow、Paddle等许多其他格式。
Tip
如果您打开导出操作下拉菜单并点击高级选项,您可以自定义每种格式的导出选项。
Note
如果您打开导出操作下拉菜单并点击高级选项,您可以重新导出每种格式。
您还可以在生产环境中使用我们的推理API。
阅读Ultralytics推理API文档以获取更多信息。
分享模型
Info
Ultralytics HUB的分享功能提供了一种便捷的方式来与他人共享模型。此功能旨在适应现有的Ultralytics HUB用户以及尚未创建账户的用户。
Note
您可以控制模型的公开访问权限。
您可以选择将公开访问设置为“私有”,在这种情况下,只有您可以访问它。或者,您可以将公开访问设置为“未列出”,这将允许任何拥有模型直接链接的人查看模型,无论他们是否拥有Ultralytics HUB账户。
导航到您想要分享的模型的模型页面,打开模型操作下拉菜单并点击分享选项。此操作将触发分享模型对话框。
Tip
您也可以直接从模型页面或包含您模型的项目页面分享模型。
将公开访问设置为“未列出”并点击保存。
现在,任何拥有模型直接链接的人都可以查看它。
Tip
您可以轻松点击分享模型对话框中显示的模型链接来复制它。
编辑模型
导航到您想要编辑的模型的模型页面,打开模型操作下拉菜单并点击编辑选项。此操作将触发更新模型对话框。
Tip
您也可以直接从模型页面或包含您模型的项目页面编辑模型。
对您的模型应用所需的修改,然后通过点击保存确认更改。
删除模型
导航到您想要删除的模型的模型页面,打开模型操作下拉菜单并点击删除选项。此操作将删除模型。
Tip
您也可以直接从模型页面或包含您模型的项目页面删除模型。
Note
如果您改变主意,可以从回收站页面恢复模型。