YOLOv9推理结果可视化：matplotlib绘图参数调整技巧

YOLOv9推理结果可视化：matplotlib绘图参数调整技巧

你已经用YOLOv9跑出了目标检测的结果，但默认的绘图效果总觉得差点意思？框太粗、字体太小、颜色不协调——别急，这其实是可视化环节没调好。本文将带你深入matplotlib的关键绘图参数，手把手教你如何优化YOLOv9推理结果的图像输出质量，让检测图更清晰、专业，适合汇报、展示甚至论文配图。

我们基于“YOLOv9官方版训练与推理镜像”环境进行操作，所有依赖已预装，无需额外配置，直接进入代码即可动手调整。

1. 可视化为何重要：不只是“看个框”

很多人跑完推理就只关心mAP和FPS，却忽略了结果可视化本身的价值。一张高质量的检测图能：

• 快速判断模型是否漏检、误检
• 展示给非技术同事或客户时更具说服力
• 用于论文、报告、PPT中提升专业感
• 帮助调试数据标注问题（比如边界框偏移）

而YOLOv9默认使用的matplotlib绘图方式虽然能用，但样式较为基础。通过微调几个关键参数，就能显著提升视觉体验。

2. 默认绘图效果分析

先来看一下YOLOv9原始detect_dual.py脚本生成的默认效果图（以horses.jpg为例）：

• 边界框线条较粗，略显笨重
• 标签文字偏小，在高分辨率图上不易看清
• 字体为默认sans-serif，缺乏设计感
• 颜色使用标准colormap，部分颜色对比度不足
• 没有边距控制，图像紧贴边缘

这些问题都可以通过修改matplotlib的绘图逻辑来解决。

2.1 找到绘图代码位置

在/root/yolov9/utils/plots.py文件中，核心绘图函数是plot_one_box()和Annotator类。我们要做的就是在保留原有功能的基础上，增强样式控制能力。

3. matplotlib绘图参数调整实战

下面我们逐项优化绘图效果，每一项都附带可运行代码片段。

3.1 调整边界框线条粗细与风格

默认情况下，边界框线宽固定为2或3，显得不够精致。我们可以根据图像尺寸动态设置线宽，并支持虚线等样式。

def plot_one_box_custom(xyxy, img, color=(128, 128, 128), label=None, line_thickness=2, line_style='-'): import matplotlib.pyplot as plt from matplotlib.patches import Rectangle # 转换坐标 x1, y1, x2, y2 = map(int, xyxy) width = x2 - x1 height = y2 - y1 # 创建图形（如果还没有） if not plt.get_fignums(): fig, ax = plt.subplots(1, figsize=(12, 8)) ax.imshow(img) else: ax = plt.gca() # 绘制矩形框 rect = Rectangle((x1, y1), width, height, linewidth=line_thickness, edgecolor=[c/255 for c in color], facecolor='none', linestyle=line_style) ax.add_patch(rect) # 添加标签（后续章节展开） if label: font_size = max(line_thickness + 3, 10) ax.text(x1, y1 - 5, label, color='white', fontsize=font_size, bbox=dict(facecolor=[c/255 for c in color], edgecolor='none', pad=2), verticalalignment='top', weight='bold')

提示：line_style可选'-'（实线）、'--'（虚线）、'-.'（点划线）、':'（点线），可用于区分不同置信度的检测框。

3.2 优化字体显示：大小、颜色与类型

默认字体小且无衬线，在复杂背景上难以阅读。我们可以通过以下方式改进：

• 动态调整字体大小
• 使用白色文字+深色背景框增强可读性
• 支持自定义字体（如SimHei中文黑体）

import matplotlib.font_manager as fm # 设置中文字体（可选） plt.rcParams['font.sans-serif'] = ['SimHei', 'Arial Unicode MS', 'DejaVu Sans'] plt.rcParams['axes.unicode_minus'] = False # 正常显示负号 # 全局字体设置 plt.rcParams.update({ 'font.size': 12, 'font.weight': 'bold', 'axes.titlesize': 16, 'axes.titleweight': 'bold' })

如果你希望在特定场景下使用更大字号：

font_settings = { 'size': max(10, int(line_thickness * 4)), 'weight': 'bold', 'family': 'sans-serif' } ax.text(x1, y1 - 5, label, fontdict=font_settings, ...)

3.3 自定义颜色方案：告别随机色块

YOLOv9默认使用随机RGB颜色，容易出现浅色系导致对比度低的问题。建议采用预定义高对比度调色盘。

# 定义专业调色板（适用于多种类别） COLOR_PALETTE = [ (0, 196, 255), # 青蓝 (255, 102, 102), # 珊瑚红 (102, 255, 102), # 浅绿 (255, 255, 102), # 柠檬黄 (255, 102, 255), # 品红 (102, 102, 255), # 蓝紫 (255, 165, 0), # 橙色 (139, 69, 19) # 棕色 ] def get_color_by_class(cls_id): return COLOR_PALETTE[cls_id % len(COLOR_PALETTE)]

这样每个类别的框都有稳定且醒目的颜色，便于长期观察和对比。

3.4 控制图像边距与布局

默认保存的图像四周太紧，影响美观。可通过bbox_inches='tight'和pad_inches控制留白。

plt.savefig('output_with_margin.jpg', dpi=300, bbox_inches='tight', pad_inches=0.1, quality=95)

• dpi=300：保证高清输出，适合打印或放大
• pad_inches=0.1：增加小幅边距，避免裁剪
• bbox_inches='tight'：自动裁剪多余空白

3.5 添加标题与统计信息

除了画框，还可以在图像上方添加全局信息，如检测数量、时间消耗等。

def add_image_title(ax, title="Detection Result", num_detections=0, inference_time=0.0): ax.set_title(f"{title}\n" f"Objects Detected: {num_detections} | " f"Inference Time: {inference_time:.3f}s", fontsize=14, pad=20, weight='bold', loc='left')

调用时：

add_image_title(plt.gca(), "YOLOv9-S Detection", 7, 0.045)

这让整张图更有“报告感”。

4. 整合优化：构建你的专属可视化函数

现在我们将上述技巧整合成一个完整的增强版绘图函数。

def enhanced_plot_detection(image_path, detections, class_names, output_path): """ 增强版YOLOv9检测结果可视化 """ img = cv2.imread(image_path) img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) plt.figure(figsize=(14, 10)) ax = plt.gca() ax.imshow(img_rgb) ax.axis('off') # 隐藏坐标轴 total_detections = 0 for det in detections: xyxy, conf, cls_id = det[:4], det[4], int(det[5]) label = f"{class_names[cls_id]} {conf:.2f}" color = get_color_by_class(cls_id) # 统一设置 line_thickness = max(2, int(conf * 3)) # 置信度越高线越粗 plot_one_box_custom(xyxy, img_rgb, color=color, label=label, line_thickness=line_thickness, ax=ax) total_detections += 1 # 添加标题 add_image_title(ax, "Enhanced YOLOv9 Detection", total_detections, 0.048) # 保存高清图 plt.savefig(output_path, dpi=300, bbox_inches='tight', pad_inches=0.1) plt.close() print(f"Enhanced visualization saved to {output_path}")

你可以将这个函数集成到detect_dual.py中替换原有绘图逻辑。

5. 实际效果对比

经过这些调整，你的检测图不再是“能看”，而是真正“好看”、“耐看”。

6. 总结

YOLOv9的强大不仅体现在检测精度上，也应体现在结果呈现的质量上。通过合理调整matplotlib的绘图参数，我们可以轻松实现：

• 更清晰的边界框显示
• 更易读的文本标签
• 更专业的整体排版
• 更适合展示的高清输出

这些改动不需要修改模型结构，也不影响推理速度，却能让最终成果的专业度大幅提升。尤其是在撰写论文、做项目汇报或向客户演示时，一张精心设计的检测图往往比十行代码更有说服力。

记住：好的AI系统，不仅要“做得准”，还要“看得清”。

获取更多AI镜像

想探索更多AI镜像和应用场景？访问 CSDN星图镜像广场，提供丰富的预置镜像，覆盖大模型推理、图像生成、视频生成、模型微调等多个领域，支持一键部署。