X
Ultralytics YOLOv8.2 发布Ultralytics YOLOv8.2 发布Ultralytics YOLOv8.2 释放箭头
绿色检查
链接复制到剪贴板

训练自定义数据集 Ultralytics YOLOv8 在 Google Colab 中

在 Google Colab 中使用Ultralytics YOLOv8 掌握自定义数据集的训练方法。从设置到培训和评估,本指南包罗万象。

Facebook 徽标Twitter 徽标LinkedIn 徽标复制链接符号

在这篇博客中,我们将了解如何使用 Ultralytics YOLOv8 在 Google Colab 中。准备好释放 YOLOv8 我们将指导您完成从设置到培训和评估的整个过程。

在 Google Colab 中设置YOLOv8 模型

让我们从在 Google Colab 中设置我们的环境开始。那么什么是 Google Colab?Google Colab 是 Google Colaboratory 的缩写,是 Google 的免费云平台,用于编写和运行 Python 法典。 

设置的第一步是选择适当的运行时类型,确保可以访问 GPU。使用 nvidia-smi 命令验证 GPU 设置,确保一切运行顺利。

接下来,使用 pip 安装Ultralytics 和YOLOv8 的依赖项。从Ultralytics 导入YOLO 模型,开始我们的自定义对象检测之旅。

标记和准备数据集

现在,让我们准备 数据集。用 边界框标注数据,为每个对象指定类别。从Ultralytics 将数据集导出为YOLOv8 格式,然后导入 Google Colab 笔记本。

培训自定义YOLOv8 模型

为物体检测设置 检测任务,并选择适合您需要的 YOLOv8 模型大小。指定数据集的位置、epochs 次数和用于训练的图像大小。借助YOLOv8 和 GPU 加速的强大功能,观看模型的学习和适应过程。

评估和验证您的模型

训练完成后,使用平均误差位置等指标评估模型的性能。在未见过的数据上验证模型,确保其泛化能力。绘制混淆矩阵并分析预测结果,以进一步微调模型。

Ultralytics YOLOv8 只需使用CLI 命令就能轻松验证模型,该命令具有自动超参数设置、多指标支持等多种关键功能。 

Ultralytics 还支持一些CLI 和Python 参数,用户可以在验证过程中根据自己的需要使用这些参数,以获得更好的输出结果。如需了解更多信息,请参阅我们的 文档

图 1. Nicolai Nielsen 概述了如何使用 Ultralytics YOLOv8 在 Google Colab 中。

让您的模型更上一层楼

现在,您已成功训练自定义 YOLOv8 Google Colab 中的模型。但我们的旅程并没有就此结束。在下一个视频中,我们将探讨如何使用经过自定义训练的模型权重导出模型权重并运行实时推理 YOLOv8 型。准备好迎接令人振奋的体验,因为我们突破了物体检测的界限。敬请关注! 

总结

感谢您加入我们,我们将了解自定义对象检测的世界 YOLOv8 在 Google Colab 中。请继续关注更多激动人心的更新和教程,我们将继续探索人工智能和机器学习的无限可能性。 

有了这个全面的指南,你现在可以训练你自己的自定义对象检测模型 Ultralytics YOLOv8 在 Google Colab 中。 在此处观看完整教程! 

让我们共同打造人工智能的未来

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

在此类别中阅读更多内容