讲师中心 微信公众号
AI工具推荐 视频效率加速

标题:YOLOv8 中仅显示分割掩码与标签(去除边界框)的完整实现教程

夜敏酱_8453

夜敏酱_8453

发布时间:2026-01-02 13:05:41

|

408人浏览过

|

来源于php中文网

原创

标题:YOLOv8 中仅显示分割掩码与标签(去除边界框)的完整实现教程

本文详解如何在 yolov8(ultralytics v8.0.132)中保存或可视化实例分割结果时,**移除边界框(boxes)但保留类别标签和置信度文本**,并提供可直接运行的自定义绘图方案。

YOLOv8 的 predict() 方法虽支持 boxes=False 来禁用边界框绘制,但官方渲染器(尤其是 segment 模式)在 boxes=False 时会连带屏蔽所有标签(labels)和置信度(conf)文本——这是由底层 Annotator 类的设计逻辑决定的:标签文本默认锚定在边界框左上角,一旦框被禁用,文本渲染即被跳过。因此,单纯依赖 show_labels=True 或 probs=True 参数组合会触发配置校验错误(如 SyntaxError: unrecognized arguments),因为这些参数在无框模式下不被 SegmentPredictor 支持。

✅ 正确解法是:绕过内置 save=True 渲染,手动提取预测结果(掩码、类别、置信度),使用 OpenCV + PIL 自定义叠加。以下是经过验证、结构清晰、开箱即用的实现方案:

YouTube Music ULTRA
YouTube Music ULTRA

使用自然语言控制YouTube Music,播放、暂停、跳过、搜索、管理播放列表和排队曲目。通过浏览器自动化实现完整的播放控制。

下载

✅ 完整可运行代码(适配 YOLOv8 分割任务)

import numpy as np
import cv2
import matplotlib.pyplot as plt
from PIL import Image
from torchvision.transforms import functional as F
from ultralytics import YOLO

# 1. 加载模型与预测
model = YOLO("path/to/best.pt")  # 替换为你的模型路径
results = model.predict("path/to/image.png", conf=0.3, save=False)  # ⚠️ 关键:save=False,避免默认渲染

# 2. 提取首张图像结果(batch size=1)
r = results[0]
if not hasattr(r, 'masks') or r.masks is None:
    print("⚠️ 未检测到分割掩码,请检查模型是否为 segmentation 类型(.pt 文件需支持 masks)")
    exit()

# 3. 加载原始图像(RGB)
img_pil = Image.open("path/to/image.png").convert("RGB")
img_np = np.array(img_pil)  # 形状: (H, W, 3)
annotated = img_np.copy()

# 4. 遍历每个掩码实例并绘制
for i, (mask_tensor, box) in enumerate(zip(r.masks.data, r.boxes)):
    # ▶️ 步骤 1:上采样掩码至原图尺寸(保持二值性)
    h, w = img_np.shape[:2]
    mask_resized = F.resize(mask_tensor.unsqueeze(0), (h, w), antialias=True).squeeze(0)
    mask_binary = (mask_resized > 0.5).cpu().numpy()  # 转为布尔 NumPy 掩码

    # ▶️ 步骤 2:应用半透明彩色覆盖(可选:更换颜色)
    color = np.array([0, 128, 255])  # BGR 格式(OpenCV 使用)→ 这里是蓝绿色
    annotated[mask_binary] = (annotated[mask_binary] * 0.5 + color * 0.5).astype(np.uint8)

    # ▶️ 步骤 3:提取标签与置信度
    cls_id = int(box.cls.item())
    conf = float(box.conf.item())
    label = f"{r.names[cls_id]}: {conf:.2f}"

    # ▶️ 步骤 4:定位文本位置(使用边界框左上角,更鲁棒;也可改用掩码质心)
    xyxy = box.xyxy.cpu().numpy()[0]
    x1, y1 = int(xyxy[0]), int(xyxy[1])
    # 防止文字超出图像边界
    x1 = max(10, min(x1, w - 150))
    y1 = max(25, min(y1, h - 10))

    # ▶️ 步骤 5:用 OpenCV 绘制白色粗体文本
    cv2.putText(
        annotated,
        label,
        (x1, y1),
        cv2.FONT_HERSHEY_SIMPLEX,
        0.7,
        (255, 255, 255),  # 白色文字
        2,
        cv2.LINE_AA
    )

# 5. 可视化结果
plt.figure(figsize=(10, 8))
plt.imshow(annotated)
plt.axis('off')
plt.title("YOLOv8 Segmentation — Masks + Labels (No Boxes)", fontsize=14, pad=15)
plt.show()

# ✅ 可选:保存结果图
Image.fromarray(annotated).save("output_no_boxes_with_labels.png")
print("✅ 已保存:output_no_boxes_with_labels.png")

? 关键说明与注意事项

  • 模型要求:确保你使用的是 YOLOv8-seg 模型(训练时指定 task=segment,导出 .pt 文件包含 masks 层)。分类(classify)或检测(detect)模型无法输出 r.masks。
  • 掩码缩放:torchvision.transforms.functional.resize 是推荐方式(支持 antialias=True),比双线性插值更保真;避免使用 cv2.resize 直接缩放浮点掩码。
  • 颜色策略:示例中使用固定色叠加(半透明),你可按 cls_id 动态分配颜色(如 plt.cm.tab10(cls_id % 10)),增强可区分性。
  • 文本定位优化:当前使用 box.xyxy 左上角,若需更精准(如贴合掩码顶部),可用 cv2.findContours + cv2.boundingRect 计算掩码最小外接矩形。
  • 性能提示:对高分辨率图像,可先缩放输入再预测(imgsz=640),或对掩码做 cv2.dilate 膨胀后绘制,提升视觉清晰度。

✅ 总结

YOLOv8 官方 API 尚未开放“仅隐藏框、保留标签”的细粒度控制选项(尤其在分割模式下)。因此,手动后处理是当前最稳定、最灵活的解决方案。本教程提供的代码已通过 Ultralytics v8.0.132 实测,兼容 Jupyter Notebook 与脚本环境,支持多类别、多实例、任意分辨率图像,并兼顾可读性与扩展性。如未来版本支持 show_boxes=False 下的独立 show_labels 开关,只需升级库并切换参数即可无缝迁移。

热门AI工具

更多
DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

花生AI
花生AI Hot

一款AI视频创作工具,主要用于B站推出的AI视频创作工具,适合需要提升相关任务效率的用户。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

相关专题

更多
Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

0

2026.11.06

Postman请求头Headers怎么设置
Postman请求头Headers怎么设置

围绕Postman HTTP请求头配置展开,讲解Headers选项的具体使用方式,包括Key和Value参数填写、请求头信息添加、常见Header字段设置及请求头修改方法,帮助用户解决接口调试过程中遇到的请求头配置问题。

0

2026.11.06

Postman接口请求发送教程
Postman接口请求发送教程

详细介绍Postman接口请求的创建与发送方法,涵盖GET和POST请求、URL配置、请求参数、Headers请求头、Body请求体设置以及响应结果查看,帮助用户掌握Postman接口请求的完整操作流程。

0

2026.11.06

DBeaver怎么导入数据
DBeaver怎么导入数据

DBeaver数据导入专题,整理Excel、CSV、SQL、TXT等常见文件的数据导入方法,涵盖目标数据库选择、数据表创建、字段映射、数据类型设置及导入参数配置等操作,帮助用户将本地文件中的数据写入数据库,并了解批量导入、重复数据处理和导入失败等常见问题的解决方法。

20

2026.11.05

DBeaver安全防护教程大全
DBeaver安全防护教程大全

DBeaver安全防护专题,围绕数据库连接安全、账号密码保护和访问权限管理展开,整理SSL/TLS加密连接、SSH隧道、数据库凭据保存及证书验证等配置方法,同时介绍敏感信息保护、远程连接风险和高权限账号使用等注意事项,帮助用户提高数据库日常管理的安全性。

20

2026.11.05

DBeaver连接创建与配置教程
DBeaver连接创建与配置教程

本专题聚焦DBeaver数据库连接配置与管理,覆盖从新建连接、选择数据库类型到填写连接参数和验证连接的完整流程。针对MySQL、PostgreSQL、SQL Server等常见数据库,整理相应的连接设置方法,并提供驱动缺失、认证失败和连接异常等问题的排查方向。

20

2026.11.04

XORM框架结构体映射与表结构同步
XORM框架结构体映射与表结构同步

本专题围绕XORM框架的结构体映射展开,说明表名与字段的命名策略、xorm标签常用属性、主键自增与时间字段自动填充,并演示Sync2自动建表、增量字段升级、索引声明与反向生成结构体的完整流程。

20

2026.11.04

XORM框架安装配置与数据库连接
XORM框架安装配置与数据库连接

本专题聚焦XORM框架的配置与连接环节,拆解MySQL、PostgreSQL、SQLite、SQL Server的DSN写法与驱动选择,讲清连接池参数、多数据源拆分、读写分离与超时重连的处理方式,并补充容器与兼容数据库的适配经验。

20

2026.11.04

XORM框架安装配置与数据库连接
XORM框架安装配置与数据库连接

本专题聚焦XORM框架的配置与连接环节,拆解MySQL、PostgreSQL、SQLite、SQL Server的DSN写法与驱动选择,讲清连接池参数、多数据源拆分、读写分离与超时重连的处理方式,并补充容器与兼容数据库的适配经验。

0

2026.11.04

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
誉天教育RHCE视频教程
誉天教育RHCE视频教程

共9课时 | 1.8万人学习

尚观Linux RHCE视频教程(二)
尚观Linux RHCE视频教程(二)

共34课时 | 7万人学习

尚观RHCE视频教程(一)
尚观RHCE视频教程(一)

共28课时 | 5.9万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn