深圳尤洛视觉
深圳
立即加入

导出一个自定义训练的 Ultralytics YOLOv8 模型

Nuvola Ladi

2 分钟阅读

2024年4月17日

使用 Ultralytics YOLOv8 探索自定义目标检测! 了解如何在网络摄像头上训练、导出和运行实时推理!

加入我们,一起仔细研究使用Ultralytics YOLOv8进行自定义对象检测。在本博文中,我们将探索训练自定义模型、导出训练后的权重以及在网络摄像头上运行实时推理的复杂过程。 

训练自定义目标检测模型

在之前的视频中,我们深入探讨了在 Google Colab 上训练 YOLOv8 模型,使用自定义杯子数据集。我们看到训练图稳步改善,损失减少,平均误差位置增加。基于此,我们特殊的模型现在可以非常准确地识别五种不同类型的杯子。

在我们的自定义模型训练完毕并准备就绪后,现在是探索下一个前沿领域的时候了——导出训练后的权重并在网络摄像头上运行实时推理。

模型推理是指使用经过训练的计算机视觉模型,基于新的、未见过的数据进行预测或决策的过程。它是指模型使用输入数据(例如图像),并通过其学习到的参数和结构对其进行处理。然后,模型根据其训练任务生成输出,例如分类、检测或分割。 

在实践中,推理通常涉及将训练好的模型部署到生产环境中,在该环境中,该模型可用于实时或批量处理场景中处理真实世界的数据。

导出训练后的模型并运行推理

通过从 Colab 下载的训练好的模型权重,我们可以无缝地将它们导入到 Python 环境中,随时可以释放我们自定义模型的全部潜力。

只需几行代码,我们就可以设置一个 Python 脚本,在网络摄像头上运行实时推理,捕捉实时画面并以惊人的准确度检测不同的杯子。YOLOv8 的强大之处在于,我们的模型可以毫不费力地识别各种形状、大小和颜色的杯子,展示了其在现实场景中的多功能性和可靠性。

微调模型具有多种优势。它们可以检测分割分类预训练模型可能不支持的对象。此外,它们还可以帮助研究人员或数据科学家了解模型架构在真实世界数据集上的表现。

__wf_reserved_inherit
图 1. Nicolai Nielsen 概述了如何训练自定义模型、导出训练后的权重以及在网络摄像头上运行实时推理。

总结

从训练自定义模型到导出训练后的权重,再到在网络摄像头上运行实时推理,我们亲身见证了 YOLOv8 的强大功能和多功能性。

加入我们,共同探索使用 Ultralytics YOLOv8 解锁自定义对象检测的全部潜力。 让我们一起塑造 AI 的未来,一次检测一个目标。查看Ultralytics HUB和我们的文档,获取关于 Ultralytics 的更多信息,并加入我们的社区,及时了解最新进展! 

如果您对导出自定义训练的YOLOv8模型并在网络摄像头上运行推理感兴趣,请观看此处的完整视频

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

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

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