使用DepthAI实现BlazePose追踪的高效方案
2024-05-21 09:01:15作者:钟日瑜

BlazePose跟踪与DepthAI是一个开源项目,它结合了Google Mediapipe的单体姿势跟踪模型和DepthAI硬件(如OAK-1,OAK-D等),为实时人体姿态识别提供了一种高性能的解决方案。
1、项目介绍
本项目旨在在DepthAI平台上运行BlazePose的各种版本(包括“full”、“lite”和“Heavy”,基于Mediapipe 0.8.6)。此外,还提供了与Mediapipe 0.8.4兼容的姿势检测模型,与上述三种地标模型兼容。项目不仅支持从外部输入源(如视频、图像)进行推理,还特别优化了内部摄像头的边缘模式,以实现实时、高效的处理速度。
2、项目技术分析
项目有两种工作模式:
- 主机模式(Host mode):主要处理在主机上进行,仅将神经网络任务交给设备,适用于外接输入源的处理。
- 边缘模式(Edge mode):大部分处理过程都在设备上完成,极大地减少了数据传输,提升了效率,是处理内部摄像头的最佳选择。
边缘模式下,不同地标模型的帧率如下:
| 地标模型 | Edge模式下的FPS | RGB_Laconic输入下的FPS |
|---|---|---|
| Full | 20 | 18 |
| Lite | 26 | 22 |
| Heavy | 8 | 7 |
此外,对于有深度感知功能的设备,项目还能实现对参考点的真三维测量。
3、项目及技术应用场景
BlazePose与DepthAI的结合适合多种应用场景:
- 健身教学:通过实时跟踪用户的动作,提供反馈或纠正错误。
- 虚拟试衣间:分析人体形状,调整虚拟服装的适配度。
- 医疗健康监测:如关节活动度测量,康复治疗监控。
- 游戏互动:游戏内角色的动作同步。
4、项目特点
- 支持多种BlazePose模型,适应不同的性能需求。
- 边缘模式优化了内部摄像头的应用,实现高速处理。
- 在有深度感应的设备上,可获取更精准的3D坐标信息。
- 提供主机和边缘两种模式,灵活性高,满足多样化场景需求。
要开始使用这个项目,只需安装Python依赖并运行demo.py脚本,通过命令行参数自定义输入源、模型和显示选项。项目详细文档提供了更多配置和调试指南。
python3 -m pip install -r requirements.txt
python3 demo.py -h
总而言之,BlazePose tracking with DepthAI是一个强大且灵活的工具,能够帮助开发者在各种应用中实现精确、快速的人体姿态识别。立即加入,开启您的创新之旅!
登录后查看全文
热门项目推荐
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0214
cann-learning-hubCANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。Jupyter Notebook0138
uni-appA cross-platform framework using Vue.jsJavaScript08
GLM-5.2智谱开源 GLM-5.2,这是针对长文本任务的最新旗舰模型。相较于前代产品 GLM-5.1,它在长文本任务处理能力上实现了显著飞跃,并且首次在稳定的 100 万 token 上下文中提供这一能力。Jinja00
SwanLab⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / LLaMA Factory / veRL/ Swift / Ultralytics / MMEngine / Keras etc.Python00
tiny-universe《大模型白盒子构建指南》:一个全手搓的Tiny-UniverseJupyter Notebook03
项目优选
收起
deepin linux kernel
C
32
16
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
469
465
暂无描述
Dockerfile
778
5.08 K
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。
C++
877
2.03 K
Ascend Extension for PyTorch
Python
758
968
本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
C++
697
1.4 K
昇腾LLM分布式训练框架
Python
185
231
本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。
C++
1.1 K
1.14 K
本仓库是 Flutter SDK 与 Flutter Engine 的 OpenHarmony 适配版本,由 CPF-Flutter 团队维护。开发者可使用熟悉的 Flutter 技术栈开发 OpenHarmony 应用,3.35.7 及以后的适配版本可基于本仓库源码构建支持 OpenHarmony 的 Flutter Engine。
Dart
1.04 K
271
JiuwenSwarm 是一款基于openJiuwen开发的智能AI Agent,它能够将大语言模型的强大能力,通过你日常使用的各类通讯应用,直接延伸至你的指尖。
Python
2.25 K
677