可视化识别结果：绘制边界框和标签的Python代码示例-编程实验室

可视化识别结果：绘制边界框和标签的Python代码示例

引言：从万物识别到可视化落地

在当前AI视觉技术快速发展的背景下，通用图像识别已成为智能应用的核心能力之一。阿里近期开源的「万物识别-中文-通用领域」模型，凭借其对中文标签的良好支持与广泛的类别覆盖，在工业检测、内容审核、智能零售等多个场景中展现出强大潜力。然而，仅仅完成推理还不够——如何将识别结果以直观、可读的方式呈现，是工程落地的关键一环。

本文聚焦于可视化识别结果的实践实现，通过一个完整的 Python 示例，手把手教你如何加载推理输出、在原图上绘制边界框（Bounding Box）与中文标签，并解决字体显示、颜色编码、文本避让等常见问题。我们将基于 PyTorch 2.5 环境，在阿里开源模型的输出基础上，构建一套可复用的可视化流程。

技术选型与环境准备

核心依赖说明

本项目运行在预设的 Conda 环境中，关键依赖如下：

# 查看/root目录下的依赖列表 cat /root/requirements.txt

典型依赖包括： -torch==2.5.0-torchvision-opencv-python-Pillow（用于中文渲染） -numpy

提示：OpenCV 默认不支持中文字符渲染，需借助 Pillow 实现中文字体绘制。

环境激活与文件操作

按照使用说明，首先激活指定环境并复制核心文件至工作区：

# 激活环境 conda activate py311wwts # 复制脚本和测试图片到工作区 cp /root/推理.py /root/workspace/ cp /root/bailing.png /root/workspace/ # 修改推理脚本中的图像路径为新位置 sed -i 's|/root/bailing.png|/root/workspace/bailing.png|g' /root/workspace/推理.py

完成上述步骤后，即可在/root/workspace目录下编辑和运行代码。

推理输出结构解析

假设推理.py已完成前向推理，返回的结果格式如下：

results = [ { 'label': '电动车', 'confidence': 0.96, 'bbox': [x_min, y_min, x_max, y_max] # 像素坐标 }, { 'label': '头盔', 'confidence': 0.89, 'bbox': [x_min, y_min, x_max, y_max] } ]

其中： -label为中文类别名； -confidence是置信度分数； -bbox为左上角与右下角坐标组成的矩形框。

该结构清晰表达了每个检测对象的空间位置与语义信息，为后续可视化提供了数据基础。

可视化实现：绘制边界框与中文标签

完整代码实现

以下是一个完整可运行的可视化函数，集成 OpenCV 与 Pillow 实现高质量中文标注：

import cv2 import numpy as np from PIL import Image, ImageDraw, ImageFont import os def draw_chinese_labels(image_path, results, font_path="simhei.ttf", output_path="output.jpg"): """ 在图像上绘制中文标签与边界框 Args: image_path: 输入图像路径 results: 检测结果列表，格式如上 font_path: 中文字体路径（需确保系统存在） output_path: 输出图像保存路径 """ # 读取图像 image = cv2.imread(image_path) if image is None: raise FileNotFoundError(f"无法加载图像: {image_path}") # 转换为RGB以便PIL处理 image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) pil_image = Image.fromarray(image_rgb) draw = ImageDraw.Draw(pil_image) # 加载中文字体（推荐使用黑体） try: font = ImageFont.truetype(font_path, 32) except IOError: # 若未找到字体，尝试使用备用路径或下载 simhei.ttf font = ImageFont.truetype("/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc", 32) # 为不同类别生成颜色（BGR格式，OpenCV使用） np.random.seed(42) colors = {} for res in results: label = res['label'] if label not in colors: colors[label] = tuple(np.random.randint(0, 255, 3).tolist()) # 绘制每一条结果 for res in results: x_min, y_min, x_max, y_max = map(int, res['bbox']) label = res['label'] conf = res['confidence'] color = colors[label] # 1. 绘制边界框（OpenCV） cv2.rectangle(image, (x_min, y_min), (x_max, y_max), color, thickness=3) # 2. 准备标签文本 text = f"{label} {conf:.2f}" # 3. 使用Pillow绘制中文标签 # 先获取文本尺寸 left, top, right, bottom = draw.textbbox((0, 0), text, font=font) text_width = right - left text_height = bottom - top # 设置标签背景位置 text_x = x_min text_y = y_min - text_height - 10 if text_y < 0: text_y = y_min + 10 # 若超出顶部，则放在框下方 # 绘制半透明背景 draw.rectangle([text_x, text_y, text_x + text_width + 10, text_y + text_height + 10], fill=tuple([int(c * 0.7) for c in color]) + (180,)) draw.text((text_x + 5, text_y + 5), text, font=font, fill=(255, 255, 255)) # 将PIL图像转回OpenCV格式 final_image = cv2.cvtColor(np.array(pil_image), cv2.COLOR_RGB2BGR) # 保存结果 cv2.imwrite(output_path, final_image) print(f"可视化结果已保存至: {output_path}") return final_image

关键技术点详解

1. 中文字体兼容性处理

OpenCV 的cv2.putText()不支持 Unicode 字符（如中文），因此必须切换至 Pillow 进行文本绘制。我们使用ImageFont.truetype()加载.ttf或.ttc字体文件。

建议字体： -simhei.ttf（黑体，Windows/Linux通用） -wqy-zenhei.ttc（文泉驿正黑，Linux常用）

若环境中无中文字体，可通过以下命令安装：

# Ubuntu/Debian apt-get install -y fonts-wqy-zenhei # 或手动上传 simhei.ttf 至项目目录

2. 颜色管理与类别区分

为提升可读性，我们为每个唯一类别分配一种随机但固定的颜色。通过 NumPy 随机种子保证相同类别的颜色一致性。

colors = {} for res in results: if res['label'] not in colors: colors[res['label']] = tuple(np.random.randint(0, 255, 3).tolist())

3. 文本避让策略

当边界框靠近图像顶部时，标签可能被截断。为此我们实现简单逻辑判断：

text_y = y_min - text_height - 10 if text_y < 0: text_y = y_min + 10 # 放在框下方

此机制确保标签始终可见。

4. 半透明背景增强可读性

直接在复杂背景上绘制白色文字易导致阅读困难。我们添加深色半透明矩形作为文字底衬：

draw.rectangle([...], fill=tuple([int(c * 0.7) for c in color]) + (180,))

Alpha 值控制透明度，使背景不过于突兀。

集成至现有推理流程

假设原始推理.py文件已有如下调用逻辑：

# 推理部分（伪代码） model = load_model() results = model.predict("bailing.png")

可在其末尾加入可视化模块：

# 可视化部分 if __name__ == "__main__": image_path = "/root/workspace/bailing.png" output_path = "/root/workspace/output_bailing.jpg" # 假设 results 来自模型输出 results = [ {'label': '电动车', 'confidence': 0.96, 'bbox': [100, 150, 300, 400]}, {'label': '骑手', 'confidence': 0.92, 'bbox': [80, 130, 320, 420]}, {'label': '头盔', 'confidence': 0.89, 'bbox': [180, 160, 240, 190]} ] draw_chinese_labels(image_path, results, output_path=output_path)

运行后将在同目录生成带标注的output_bailing.jpg。

常见问题与解决方案

| 问题 | 原因 | 解决方案 | |------|------|----------| | 中文显示为方框或乱码 | 缺少中文字体或路径错误 | 安装simhei.ttf并正确传入路径 | | 图像保存失败 | 输出路径权限不足或目录不存在 | 使用os.makedirs(os.path.dirname(output_path), exist_ok=True)创建目录 | | 边界框颜色混乱 | 每次都重新生成颜色 | 固定随机种子或按类别缓存颜色 | | 标签重叠遮挡 | 多个目标密集分布 | 引入非极大抑制（NMS）预处理或动态调整文本偏移 |