1. 介绍
在之前的文章中,介绍了如何使用 YOLOv8 在不同的编程语言来检测图片中的对象。然而,YOLOv8 还可以把检测到的目标图像分割出来,本篇文章将介绍如何使用YOLOv8做图片分割。
对象检测的结果是所有检测到的对象的边界框。图像分割的结果是所有检测到的对象的蒙版。它是一个黑白图像,其中属于该对象的所有像素都是白色,所有其他像素都是黑色,如下图所示:

得到此蒙版(实际上是一个 2D 数组,其中背景像素值为 0,对象像素值为 255)后,可以将其应用于原始图像,仅绘制在蒙版上显示为白色的像素区域。
例如,可以通过这种方式从图像中删除背景:

并设置新背景:
同样,可以对视频中的每个帧运行实例分割,并从整个视频中删除背景。
此外,有了检测到的对象的这些蒙版,可以计算它们的轮廓。
本文详细介绍了如何使用YOLOv8进行图像分割,包括使用官方预训练模型、创建自定义训练数据、导出模型到ONNX以及在生产环境中应用。文章通过代码示例展示了从获取分割掩码到绘制轮廓多边形的全过程。
订阅专栏 解锁全文

2万+

被折叠的 条评论
为什么被折叠?



