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

在 Google Colab 上使用 Ultralytics YOLO11 进行图像分割

Abirami Vina

4 分钟阅读

2024年12月30日

了解如何有效地使用 Ultralytics YOLO11 进行图像分割,利用 Google Colab 上的汽车零件数据集进行无缝训练和测试。

Ultralytics YOLO 模型,如最新的 Ultralytics YOLO11,支持各种计算机视觉任务,例如对象检测、图像分类和实例分割。这些任务中的每一个都旨在复制人类视觉的特定方面,使机器能够看到和解释周围的世界。 

例如,想象一下艺术课上的学生如何拿起铅笔并勾勒出图画中的物体。在幕后,他们的大脑正在执行分割——将对象与背景和其他元素区分开。图像分割使用人工智能 (AI) 实现类似的目标,将视觉数据分解为有意义的部分,供机器理解。这项技术可以应用于许多行业的各种应用中。 

__wf_reserved_inherit
图 1. Ultralytics YOLO11 用于分割图像中的对象。

一个实际的例子是汽车零件分割。通过识别和分类车辆的特定组件,图像分割可以简化汽车制造、维修和电子商务目录等行业的流程。

在本文中,我们将探讨如何使用 Ultralytics YOLO11、Google Colab 和 Roboflow Carparts Segmentation 数据集构建一个能够准确识别和分割汽车零件的解决方案。

Ultralytics YOLO11 易于使用

Ultralytics YOLO11 作为一个预训练模型提供,该模型在 COCO 数据集上训练,涵盖 80 个不同的对象类别。但是,对于特定的应用,例如分割汽车零件,可以对模型进行自定义训练,以更好地适应您的数据集和用例。这种灵活性使 YOLO11 能够在通用和高度专业的任务中表现良好。

自定义训练涉及使用预训练的 YOLO11 模型,并根据新的数据集对其进行微调。通过提供特定于您任务的标记示例,模型可以学习识别和分割您项目中独有的对象。与依赖通用的预训练权重相比,自定义训练可确保更高的准确性和相关性。

设置 YOLO11 进行自定义训练非常简单。只需最少的设置,您就可以加载模型和数据集,开始训练,并在训练过程中监控损失和准确性等指标。YOLO11 还包括用于验证和评估的内置工具,从而可以更轻松地评估模型的性能。 

在 Google Colab 上运行 Ultralytics YOLO11

在自定义训练 YOLO11 时,有几种不同的环境设置选项。其中一个最容易访问和最方便的选择是 Google Colab。以下是在 YOLO11 训练中使用 Google Colab 的一些优势:

  • 免费访问资源: Google Colab 提供 GPU(图形处理单元)和 TPU(张量处理单元),使您无需昂贵的硬件即可训练 YOLO11。
  • 协作环境: Google Colab 帮助您共享 notebook,在 Google Drive 中存储工作,并通过简单的协作和版本跟踪简化团队合作。
  • 预安装的库: 借助 PyTorch 和 TensorFlow 等预安装的工具,Google Colab 简化了设置过程,并帮助您快速入门。
  • 云集成: 您可以轻松地从 Google Drive、GitHub 或其他云来源加载数据集,从而简化了数据准备和存储。
__wf_reserved_inherit
图 2. Google Colab YOLO11 notebook。

Ultralytics 还提供了一个预配置的 Google Colab notebook,专门用于 YOLO11 训练。此 notebook 包括您需要的一切,从模型训练到性能评估,使整个过程简单易懂。这是一个很好的起点,让您可以专注于为您的特定需求微调模型,而无需担心复杂的设置步骤。

Roboflow Carparts Segmentation 数据集概述

在确定训练环境后,下一步是收集数据或选择合适的数据集来分割汽车零件。Roboflow Carparts Segmentation 数据集可在 Roboflow Universe 上获得,由 Roboflow 维护,Roboflow 是一个提供构建、训练和部署计算机视觉模型工具的平台。该数据集包括 3,156 张训练图像、401 张验证图像和 276 张测试图像,所有图像都具有高质量的汽车零件标注,例如保险杠、车门、后视镜和车轮。

通常,您需要从 Roboflow Universe 下载数据集,并手动设置它以在 Google Collab 上进行训练。但是,Ultralytics Python 包通过提供无缝集成和预配置的工具来简化此过程。

__wf_reserved_inherit
图 3. 汽车零件分割数据集中的示例。

借助 Ultralytics,可以通过预配置的 YAML 文件使用该数据集,该文件包括数据集路径、类标签和其他训练参数。这为您完成了设置,因此您可以快速加载数据集并直接开始训练模型。此外,该数据集的结构包含专门的训练集、验证集和测试集,从而可以更轻松地监控进度和评估性能。

通过利用 Roboflow Carparts Segmentation 数据集和 Ultralytics YOLO11 提供的工具,您可以获得一个无缝的工作流程,以便在 Google Colab 等平台上高效地构建分割模型。这种方法减少了设置时间,并使您可以专注于为实际应用改进模型。

汽车零件分割的实际应用

汽车零件分割在不同的行业中具有各种实际用途。例如,在维修店中,它可以帮助快速识别和分类损坏的组件,从而使维修过程更快、更高效。同样,在保险行业中,分割模型可以通过分析损坏车辆的图像以识别受影响的零件来自动进行理赔评估。这加快了理赔过程,减少了错误,并为保险公司和客户节省了时间。

__wf_reserved_inherit
图 4. 使用 YOLO 分割汽车零件。

在制造方面,分割通过检查汽车零件的缺陷、确保一致性和减少浪费来支持质量控制。这些应用展示了汽车零件分割如何通过使流程更安全、更快和更准确来改变行业。

分步指南:在 Google Colab 上使用 YOLO11 

现在我们已经介绍了所有的细节,是时候将它们整合在一起了。首先,您可以查看我们的YouTube视频,它将指导您完成设置、训练和验证用于汽车零件分割的YOLO11模型的整个过程。

以下是所涉及步骤的快速概述:

  • 在Google Colab上设置您的环境:启用GPU支持并安装Ultralytics Python包,为模型训练做准备。
  • 加载YOLO11模型:从预训练的YOLO11分割模型开始,以节省时间并利用现有功能进行汽车零件分割。
  • 使用数据集训练模型:在训练期间使用“carparts-seg.yaml”文件自动下载、配置和使用Roboflow Carparts Segmentation Dataset(汽车零件分割数据集)。调整epochs(周期)、图像大小和batch size(批次大小)等参数以微调模型。
  • 监控训练进度:跟踪关键性能指标,例如分割损失和平均精度均值(mAP),以确保模型按预期改进。
  • 验证和部署模型:在验证集上测试训练后的模型,以确认其准确性,并将其导出用于实际应用,如质量控制或保险理赔处理。

使用YOLO11进行汽车零件分割的优势

YOLO11是用于汽车零件分割的可靠且高效的工具,它提供了一系列优势,使其成为各种实际应用的理想选择。以下是主要优势:

  • 速度和效率: YOLO11可以快速处理图像,同时保持高精度,使其适用于实时任务,如质量控制和自动驾驶汽车。
  • 高精度:该模型擅长检测和分割单个图像中的多个对象,确保精确识别汽车零件。
  • 可扩展性:YOLO11可以处理大型数据集和复杂的分割任务,使其可扩展用于工业应用。
  • 多种集成:Ultralytics支持与Google Colab、Ultralytics Hub和其他常用工具等平台集成,从而增强了开发人员的灵活性和可访问性。

在Google Colab上使用YOLO11的技巧

虽然Google Colab使机器学习工作流程更容易处理,但如果您是新手,可能需要一些时间来适应。 浏览基于云的设置、运行时设置和会话限制起初可能会让人感到棘手,但有一些技巧可以使事情变得更加顺利。

以下是一些需要记住的注意事项:

  • 首先在运行时设置中启用GPU加速以加快训练速度。
  • 由于Colab在云中运行,请确保您具有稳定的互联网连接,以便访问数据集和存储库等资源。
  • 在Google Drive或GitHub中组织您的文件和数据集,使其易于在Colab中加载和管理。
  • 如果您在Colab的免费层上遇到内存限制,请尝试在训练期间减小图像大小或批次大小。
  • 请记住定期保存您的模型和结果,因为Colab会话有时间限制,您不希望丢失进度。 

使用YOLO11实现更多目标

Ultralytics YOLO11与Google Colab等平台以及Roboflow Carparts Segmentation dataset(汽车零件分割数据集)等数据集相结合,使图像分割变得简单易用。 凭借其直观的工具、预训练模型和简易设置,YOLO11使您可以轻松地投入到高级计算机视觉任务中。 

无论您是提高汽车安全性、优化制造还是构建创新的AI应用程序,这种组合都提供了帮助您成功的工具。 借助Ultralytics YOLO11,您不仅仅是在构建模型,而是在为现实世界中更智能、更高效的解决方案铺平道路。

要了解更多信息,请查看我们的GitHub存储库,并与我们的社区互动。 在我们的解决方案页面上探索自动驾驶汽车农业计算机视觉中的AI应用。 🚀

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

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

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