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

努沃拉-拉迪

2 分钟阅读

2024年4月11日

在 Google Colab 中使用 Ultralytics YOLOv8 训练自定义数据集。从设置到培训和评估,本指南涵盖了所有内容。

在本博客中,我们将介绍如何在 Google Colab 中使用 Ultralytics YOLOv8掌握自定义对象检测。我们将指导您完成从设置到培训和评估的整个过程,准备好释放 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 参数,用户可以根据自己的需要在验证过程中使用这些参数,以获得更好的输出结果。更多信息,请参阅我们的 文档

__wf_保留继承
图 1.Nicolai Nielsen简要介绍了如何在 Google Colab 中使用 Ultralytics YOLOv8 训练自定义数据集。

让您的模型更上一层楼

现在,您已经在 Google Colab 中成功训练了自定义 YOLOv8 模型。但我们的旅程并没有就此结束。在下一个视频中,我们将探索如何导出模型权重,并使用自定义训练的 YOLOv8 模型运行实时推理。我们将不断挑战物体检测的极限,为您带来激动人心的体验。敬请期待! 

总结

感谢您加入我们的行列,与我们一起在 Google Colab 中使用 YOLOv8 探索自定义对象检测的世界。我们将继续探索人工智能和机器学习的无限可能性,敬请期待更多精彩更新和教程。 

有了这份全面的指南,您就可以在 Google Colab 中使用 Ultralytics YOLOv8 训练自己的自定义对象检测模型了。 点击此处观看完整教程! 

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

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

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