YOLOv8 分割任务中获取预测实例类别名称的教程

本教程详细指导如何在 yolov8 分割任务中，从预测结果中程序化地获取每个检测到的分割实例的类别名称。我们将重点讲解如何通过访问 `results` 对象的 `boxes.cls` 属性来获取类别索引，并结合 `model.names` 映射将其转换为可读的类别名称，从而精确识别每个分割对象的类别。

在使用 YOLOv8 进行目标分割时，我们经常需要识别每个分割出的对象的具体类别。虽然 model.predict() 方法在可视化输出中能够正确区分不同类别的实例，但通过程序化方式从 result[0].masks.data 中直接获取类别信息可能会遇到困惑，因为 masks 对象本身并不直接包含类别属性。本教程将阐明如何正确地从 YOLOv8 的预测结果中提取每个分割实例的类别名称。

核心原理：类别信息的位置

在 YOLOv8 的 Results 对象中，尽管 masks 属性提供了分割掩码数据，但与每个分割实例对应的类别信息实际上存储在 boxes 属性中。boxes 属性包含了边界框、置信度以及最重要的——类别索引（cls）。YOLOv8 确保了 masks 和 boxes 属性中的检测结果是按相同顺序排列的，这意味着我们可以通过相同的索引同时访问对应实例的分割掩码和类别信息。

实现步骤与示例代码

以下是获取 YOLOv8 分割结果中实例类别名称的详细步骤及相应的 Python 代码：

标签： python pytorch 应用开发 ultra 排列 red