AI人体骨骼检测结果美化：自定义颜色线条样式教程

AI人体骨骼检测结果美化：自定义颜色线条样式教程

1. 引言：AI 人体骨骼关键点检测的可视化挑战

随着人工智能在计算机视觉领域的深入发展，人体姿态估计（Human Pose Estimation）已成为智能健身、动作捕捉、虚拟试衣等应用的核心技术之一。Google 的MediaPipe Pose模型凭借其高精度、低延迟和轻量化设计，成为当前最受欢迎的姿态检测方案之一。

然而，在实际应用中，尽管 MediaPipe 提供了默认的骨骼可视化效果（红点+白线），但这种“火柴人”风格的展示方式在专业场景下显得单调且缺乏辨识度。例如，在教学视频标注、运动分析报告或交互式艺术装置中，我们往往需要更具视觉表现力的骨架呈现方式——比如自定义关节颜色、调整连接线粗细、使用渐变色表示动作强度等。

本文将带你深入探索如何对 MediaPipe 输出的骨骼关键点进行可视化美化，通过修改绘制参数与扩展渲染逻辑，实现个性化、高可读性的骨骼连线样式。无论你是开发者、设计师还是AI爱好者，都能快速掌握这一实用技能。

2. 技术基础：MediaPipe Pose 核心能力解析

2.1 关键点定位机制

MediaPipe Pose 基于 BlazePose 架构，能够在单帧图像中精准识别33 个 3D 骨骼关键点，涵盖：

• 面部特征点（如眼睛、耳朵）
• 上肢结构（肩、肘、腕）
• 下肢结构（髋、膝、踝）
• 躯干与脊柱关键节点

这些关键点以(x, y, z, visibility)四元组形式输出，其中visibility表示该点是否被遮挡，为后续可视化提供可靠性判断依据。

2.2 默认可视化方案局限性

MediaPipe 自带的绘图工具mp_drawing提供了便捷的draw_landmarks()方法，但它存在以下限制：

• 颜色固定：关节为红色，连接线为白色
• 样式单一：线条粗细不可调，无动态效果
• 缺乏语义区分：所有骨骼连接统一处理，无法突出特定部位（如手臂 vs 腿部）

这使得其输出难以满足多样化应用场景的需求。

3. 实践应用：自定义骨骼颜色与线条样式的完整实现

3.1 技术选型与改造思路

为了突破默认样式的限制，我们需要绕过mp_drawing.draw_landmarks()的封装，转而使用底层 API 手动控制每一条骨骼的绘制行为。

我们选择手动绘制 + 封装优化的方式，兼顾灵活性与可维护性。

3.2 核心代码实现

以下是完整的 Python 实现代码，包含自定义颜色映射、线宽调节和关键点过滤逻辑：

import cv2 import mediapipe as mp import numpy as np # 初始化模型 mp_pose = mp.solutions.pose mp_drawing = mp.solutions.drawing_utils pose = mp_pose.Pose(static_image_mode=False, model_complexity=1, enable_segmentation=False) # 自定义颜色映射表（BGR格式） COLOR_MAP = { 'left_arm': (255, 0, 0), # 蓝色 - 左臂 'right_arm': (0, 255, 0), # 绿色 - 右臂 'left_leg': (0, 0, 255), # 红色 - 左腿 'right_leg': (255, 255, 0), # 青色 - 右腿 'torso': (128, 0, 128) # 紫色 - 躯干 } # 自定义连接关系分组 ARM_CONNECTIONS = [(11,13), (13,15)] # 左肩→肘→腕 RIGHT_ARM_CONNECTIONS = [(12,14), (14,16)] LEG_CONNECTIONS = [(23,25), (25,27)] RIGHT_LEG_CONNECTIONS = [(24,26), (26,28)] TORSO_CONNECTIONS = [(11,23), (12,24), (11,12)] def draw_custom_skeleton(image, landmarks): h, w, _ = image.shape # 绘制左臂（蓝色） for start_idx, end_idx in ARM_CONNECTIONS: start = landmarks.landmark[start_idx] end = landmarks.landmark[end_idx] if start.visibility > 0.5 and end.visibility > 0.5: x1, y1 = int(start.x * w), int(start.y * h) x2, y2 = int(end.x * w), int(end.y * h) cv2.line(image, (x1, y1), (x2, y2), COLOR_MAP['left_arm'], thickness=4) # 绘制右臂（绿色） for start_idx, end_idx in RIGHT_ARM_CONNECTIONS: start = landmarks.landmark[start_idx] end = landmarks.landmark[end_idx] if start.visibility > 0.5 and end.visibility > 0.5: x1, y1 = int(start.x * w), int(start.y * h) x2, y2 = int(end.x * w), int(end.y * h) cv2.line(image, (x1, y1), (x2, y2), COLOR_MAP['right_arm'], thickness=4) # 绘制左腿（红色） for start_idx, end_idx in LEG_CONNECTIONS: start = landmarks.landmark[start_idx] end = landmarks.landmark[end_idx] if start.visibility > 0.5 and end.visibility > 0.5: x1, y1 = int(start.x * w), int(start.y * h) x2, y2 = int(end.x * w), int(end.y * h) cv2.line(image, (x1, y1), (x2, y2), COLOR_MAP['left_leg'], thickness=5) # 绘制右腿（青色） for start_idx, end_idx in RIGHT_LEG_CONNECTIONS: start = landmarks.landmark[start_idx] end = landmarks.landmark[end_idx] if start.visibility > 0.5 and end.visibility > 0.5: x1, y1 = int(start.x * w), int(start.y * h) x2, y2 = int(end.x * w), int(end.y * h) cv2.line(image, (x1, y1), (x2, y2), COLOR_MAP['right_leg'], thickness=5) # 绘制躯干（紫色） for start_idx, end_idx in TORSO_CONNECTIONS: start = landmarks.landmark[start_idx] end = landmarks.landmark[end_idx] if start.visibility > 0.5 and end.visibility > 0.5: x1, y1 = int(start.x * w), int(start.y * h) x2, y2 = int(end.x * w), int(end.y * h) cv2.line(image, (x1, y1), (x2, y2), COLOR_MAP['torso'], thickness=6) # 绘制关键点（白色圆圈） for idx, landmark in enumerate(landmarks.landmark): if landmark.visibility > 0.5: cx, cy = int(landmark.x * w), int(landmark.y * h) cv2.circle(image, (cx, cy), radius=5, color=(255, 255, 255), thickness=-1) # 示例主程序 image_path = "input.jpg" image = cv2.imread(image_path) rgb_image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) results = pose.process(rgb_image) if results.pose_landmarks: draw_custom_skeleton(image, results.pose_landmarks) cv2.imwrite("output_custom_skeleton.jpg", image)

3.3 代码解析与关键技巧

✅ 分组绘制策略

通过将骨骼连接按身体区域划分（如手臂、腿部），我们可以为不同部位分配独立的颜色和线宽，增强视觉层次感。

✅ 可见性过滤

利用landmark.visibility字段避免绘制被遮挡的关键点，防止出现错误连线。

✅ BGR色彩空间适配

OpenCV 使用 BGR 色彩空间，因此(255, 0, 0)实际对应蓝色，需注意与 RGB 的区别。

✅ 线条加粗提升可读性

将线宽设置为4~6像素，确保在小尺寸图像或远距离观看时仍清晰可见。

4. 进阶优化建议与常见问题解决

4.1 动态样式增强

可以进一步引入动态效果来提升表现力：

• 根据动作幅度调整颜色深浅：例如弯曲角度越大，线条越红
• 添加渐变过渡：使用插值算法实现从起点到终点的颜色渐变
• 闪烁提示异常姿势：当检测到不平衡动作时，让相关骨骼短暂闪烁

# 示例：基于角度变化调整颜色强度 def get_color_by_angle(angle, min_angle=90, max_angle=180): norm = np.clip((angle - min_angle) / (max_angle - min_angle), 0.0, 1.0) intensity = int(255 * norm) return (0, 0, intensity) # 越直越蓝，越弯越红

4.2 WebUI 集成注意事项

若你正在使用本项目提供的 WebUI 环境，请注意：

• 修改后的脚本需替换原始inference.py或app.py中的绘图函数
• 确保 OpenCV 和 MediaPipe 版本兼容（推荐使用opencv-python==4.8.*,mediapipe==0.10.*）
• 若出现内存泄漏，可在每次推理后调用pose.close()并重新初始化

4.3 常见问题 FAQ

Q: 为什么某些连接线没有显示？
A: 检查visibility是否低于阈值 0.5，可能是关键点被遮挡或模型未准确定位。

Q: 如何导出透明背景的骨骼图？
A: 创建一个全黑或全透明画布，仅绘制骨骼线，最后保存为 PNG 格式。

Q: 能否支持多人检测？
A: MediaPipe 支持多姿态检测（pose = mp_pose.Pose(..., min_detection_confidence=0.5)），只需循环处理每个pose_landmarks即可。

5. 总结

本文系统讲解了如何对 MediaPipe Pose 输出的人体骨骼关键点进行深度可视化美化，主要内容包括：

1. 理解默认绘图机制的局限性，明确自定义需求；
2. 掌握手动绘制骨骼连接的技术路径，通过 OpenCV 实现精细控制；
3. 实践分区域着色、线宽调节、可见性过滤等核心技巧；
4. 提供可运行的完整代码示例，并给出 WebUI 部署与性能优化建议。

通过这套方法，你可以轻松将“火柴人”升级为具有专业美感的动态骨架图，广泛应用于运动分析、舞蹈教学、AR互动等领域。

未来还可结合姿态分类器、动作追踪算法，构建更智能的视觉分析系统。

💡获取更多AI镜像

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