Ultralytics YOLO27: 加入候补名单。
获取 YOLO 许可
Ultralytics YOLO

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

了解如何逐步在 Google Colab 中使用 Carparts Segmentation 数据集训练和测试 Ultralytics YOLO11 分割模型。

Abirami VinaAbirami Vina6 分钟阅读
了解如何逐步在 Google Colab 中使用 Carparts Segmentation 数据集训练和测试 Ultralytics YOLO11 分割模型。

Ultralytics YOLO 模型,例如 Ultralytics YOLO11 及其后续版本 Ultralytics YOLO26,支持多种计算机视觉任务,如目标检测、实例分割和图像分类。每种任务都旨在复现人类视觉的某个特定方面,让机器能够观察并理解周围的世界。

例如,想象美术课上的学生拿起铅笔,在画作中勾勒出一个物体。学生的大脑在幕后进行着分割:将物体与背景及其他元素区分开来。图像分割借助人工智能 (AI) 实现类似目标,将视觉数据拆解为有意义的部分,方便机器理解。这项技术可用于许多行业的各种应用。

Ultralytics YOLO11 对图像中的对象进行分割

图 1。使用 Ultralytics YOLO11 对图像中的对象进行分割。

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

本文将介绍如何使用 Ultralytics YOLO11、Google Colab 和 Roboflow Carparts Segmentation 数据集,构建能够准确识别和分割汽车零部件的解决方案。

Ultralytics YOLO11 易于使用#

Ultralytics YOLO11 提供基于 COCO 数据集训练的预训练模型,涵盖 80 个不同的对象类别。不过,对于汽车零部件分割等特定应用,可以使用自定义数据集和用例对模型进行定制训练。得益于这种灵活性,YOLO11 在通用任务和高度专业化任务中都能表现出色。

自定义训练是指使用预训练的 Ultralytics YOLO11 模型,并在新数据集上对其进行微调。通过提供与你的任务相关的标注示例,模型可以学会识别和分割项目中特有的物体。与依赖通用预训练权重相比,自定义训练通常能带来更高的精度和相关性。

设置 Ultralytics YOLO11 以进行自定义训练非常简单。只需进行少量设置,你就可以加载模型和数据集、开始训练,并在训练过程中监控损失和准确率等指标。YOLO11 还内置了验证和评估工具,让你更容易评估模型的表现。

在 Google Colab 上运行 Ultralytics YOLO11#

自定义训练 Ultralytics YOLO11 时,有几种不同的环境配置方式。Google Colab 是最容易上手、最方便的选择之一。以下是使用 Google Colab 训练 YOLO11 的一些优势:

  • 免费使用计算资源:Google Colab 提供图形处理器 (GPU)和 TPU(张量处理单元),让你无需昂贵的硬件就能训练 Ultralytics YOLO11。
  • 协作环境:Google Colab 支持你共享笔记本、将工作保存在 Google Drive 中,并通过便捷的协作功能和版本跟踪简化团队协作。
  • 预装库:Google Colab 预装了 PyTorch 和 TensorFlow 等工具,简化了环境配置流程,让你可以快速开始。
  • 云集成:你可以轻松从 Google Drive、GitHub 或其他云端来源加载数据集,从而简化数据准备和存储。

Google Colab 上的 Ultralytics YOLO11 笔记本

图 2。 Google Colab 上的 Ultralytics YOLO11 笔记本。

Ultralytics 还提供专为 YOLO11 训练设计的预配置 Google Colab 笔记本。笔记本包含从模型训练到性能评估所需的一切,让整个流程简单明了、易于操作。这是一个很好的起点,你可以专注于针对特定需求微调模型,而不必担心复杂的配置步骤。

Roboflow Carparts Segmentation 数据集概览#

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

通常,你需要从 Roboflow Universe 下载数据集,并手动设置,以便在 Google Colab 上进行训练。不过,Ultralytics Python 软件包提供了内置的数据集配置功能,简化了这一流程。

汽车零部件分割数据集示例

图 3。汽车零部件分割数据集示例。

使用 Ultralytics 时,可以通过预配置的 YAML 文件直接使用数据集。该文件包含数据集路径、类别标签和其他训练参数。你无需自行处理这些设置,因此可以快速加载数据集并直接开始训练模型。此外,数据集按训练集、验证集和测试集分别组织,便于监控进度和评估性能。

借助 Roboflow Carparts Segmentation Dataset 和 Ultralytics YOLO11 提供的工具,你可以在 Google Colab 等平台上使用简化的工作流程,高效构建分割模型。这种方法缩短了设置时间,让你能专注于优化模型,以便将其用于实际应用。

汽车零部件分割的实际应用#

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

使用 YOLO 分割汽车零部件

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

在制造领域,分割技术可以通过检查汽车零部件缺陷来支持质量控制,确保产品一致性并减少浪费。这些应用展示了汽车零部件分割如何让流程更安全、更快捷、更准确,从而推动行业转型。

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

现在我们已经介绍了所有细节,是时候把它们整合起来了。要开始操作,可以观看我们的 YouTube 视频,视频将引导你完成整个流程,包括设置、训练和验证用于汽车零部件分割的 Ultralytics YOLO11 模型。

以下是相关步骤的简要概览:

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

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

Ultralytics YOLO11 是一款可靠、高效的汽车零部件分割工具,具有多项优势,适用于各种实际应用。以下是其主要优势:

  • 速度和效率:Ultralytics YOLO11 能够快速处理图像,同时保持高准确率,因此适用于质量控制和自动驾驶汽车等实时任务。
  • 高准确率:该模型擅长检测和分割单张图像中的多个对象,确保准确识别汽车零部件。
  • 可扩展性:Ultralytics YOLO11 能够处理大型数据集和复杂的分割任务,因此适合扩展到工业应用。
  • 多种集成选项:Ultralytics 支持与 Google Colab、Ultralytics HUB(现为 Ultralytics Platform)等平台及其他热门工具集成,提升了开发者的灵活性和易用性。

在 Google Colab 上使用 Ultralytics YOLO11 的技巧#

虽然 Google Colab 能让机器学习工作流程更容易处理,但如果你是新手,可能需要一些时间来适应。刚开始时,基于云端的环境配置、运行时设置和会话限制可能让人觉得棘手,不过有几个技巧可以让整个过程顺畅得多。

以下是一些需要牢记的事项:

  • 首先,在运行时设置中启用 GPU 加速,以加快训练速度。
  • 由于 Colab 在云端运行,请确保网络连接稳定,以便访问数据集和代码仓库等资源。
  • 将文件和数据集整理到 Google Drive 或 GitHub 中,方便在 Colab 中加载和管理。
  • 如果 Colab 免费版的内存容量有限,可以尝试在训练期间减小图像尺寸或批次大小。
  • 记得定期保存模型和结果,因为 Colab 会话有时间限制,你肯定不想丢失进度。

借助 Ultralytics YOLO11,成就更多#

Ultralytics YOLO11 与 Google Colab 等平台以及 Roboflow Carparts Segmentation 数据集等数据集相结合,让图像分割变得简单易用。YOLO11 拥有直观的工具、预训练模型和简便的设置,让你轻松开展高级计算机视觉任务。

无论你是在提升汽车安全、优化制造流程,还是开发创新 AI 应用,这种组合都能提供助你成功的工具。借助 Ultralytics YOLO11,你可以为现实问题构建实用、高效的解决方案。

想了解更多,请查看我们的 GitHub 仓库,并参与社区交流。你还可以在我们的解决方案页面中探索 AI 在自动驾驶汽车和农业计算机视觉领域的应用。

作者

Abirami Vina
Abirami Vina

内容创作者 · Ultralytics

Abirami Vina 职业生涯始于计算机视觉工程师,由此在视觉 AI 和机器学习领域打下了坚实基础。她是 Ultralytics 的内容创作者,也是 Scribe of AI 的创始人兼首席撰稿人,为专注 AI 的公司撰写高质量技术内容。她热衷于让 AI 更易理解、更具吸引力,通过自己的作品连接前沿技术与现实影响。

评论

提出问题、分享你的结果,或帮助其他人。

探索解决方案

体育领域的计算机视觉

体育领域的计算机视觉

了解 Ultralytics YOLO 如何通过球员和球追踪、表现分析、教练洞察及运动员康复,推动体育视觉技术的发展。
了解详情
航空影像计算机视觉

航空影像计算机视觉

使用 Ultralytics YOLO 将无人机和航空影像转化为农业、水产养殖、保护工作和地理空间分析的实时洞察。
了解详情
航空航天中的计算机视觉

航空航天中的计算机视觉

使用 Ultralytics YOLO 将计算机视觉应用于航空航天领域,包括飞机检查、跑道 FOD 检测、机场地面作业和复合材料缺陷检测。
了解详情
安防领域的计算机视觉

安防领域的计算机视觉

使用 Ultralytics YOLO 模型守护每个场所。视觉 AI 可用于周界监控、威胁检测、车牌识别和工作场所安全。
了解详情
机器人领域的计算机视觉

机器人领域的计算机视觉

使用 Ultralytics YOLO 模型提升智能机器的能力。机器人领域的视觉 AI 可用于自主导航、感知、目标跟踪和实时控制。
了解详情
物流中的计算机视觉

物流中的计算机视觉

借助 Ultralytics YOLO 模型优化物流流程。视觉 AI 可用于包裹检查、分拣、车辆跟踪和仓库安全实时监控。
了解详情
零售中的计算机视觉

零售中的计算机视觉

借助 Ultralytics YOLO 模型重新构想零售。视觉 AI 可用于库存跟踪、货架监控、排队管理和更深入的客户洞察。
了解详情
医疗领域的计算机视觉

医疗领域的计算机视觉

借助 Ultralytics YOLO 模型构建医疗解决方案。医疗领域的视觉 AI 可加快医学影像处理、优化诊断并监测患者。
了解详情
制造业中的计算机视觉

制造业中的计算机视觉

利用 Ultralytics YOLO 模型优化制造流程。视觉 AI 可用于质量控制、缺陷检测、个人防护装备合规检查和装配线自动化。
了解详情
汽车领域的计算机视觉

汽车领域的计算机视觉

利用 Ultralytics YOLO 模型在汽车领域应用计算机视觉。视觉 AI 可提升道路安全、驾驶辅助和车辆自动化水平,让道路更智能。
了解详情
农业中的计算机视觉

农业中的计算机视觉

借助 Ultralytics YOLO 模型,将视觉 AI 引入智慧农业。为作物监测、牲畜追踪和精准农业提供支持,从而提高产量并实现更智能的生产。
了解详情

让我们携手构建 AI 的未来!

开启你的旅程,迎接机器学习的未来