MMPose 姿态估计工具箱:一张照片标出 17 到 133 个关键点
【免费下载链接】mmposeOpenMMLab Pose Estimation Toolbox and Benchmark.项目地址: https://gitcode.com/GitHub_Trending/mm/mmpose
MMPose 是 OpenMMLab 开源的姿态估计工具箱,基于 PyTorch,支持 2D/3D 人体、手部、人脸、动物与全身关键点检测,适合做人体姿态估计、动作分析的开发者直接上手。
它到底能做什么
如果你要标出照片里一个人的 17 个身体关节点,MMPose 的 RTMPose 系列能直接输出每个点的坐标和置信度。
如果你要给一段视频做实时动作分析,它同时提供自顶向下(top-down)和自底向上(bottom-up)两条路线,前者单人精度高,后者适合一次标出多人。
如果对象不是人——动物、手势、面部关键点,甚至 133 点的全身关键点,它都内置了对应的数据集适配器和预训练模型,不用自己搭数据管道。
从零跑通:MMPose 姿态估计工具箱最短路径安装
克隆仓库并安装(需要 PyTorch 1.8+):
git clone https://gitcode.com/GitHub_Trending/mm/mmpose cd mmpose && pip install -r requirements.txt && pip install -v -e .这一条把依赖和框架本体都装好,-e .表示以可编辑模式装源码,改代码即时生效。
再跑一条推理,human是预置的人体姿态模型别名,会自动配好人检模型:
python demo/inferencer_demo.py tests/data/coco/000000000785.jpg --pose2d 'human' --show --pred-out-dir predictions这一步对一张滑雪图做姿态估计,--show弹出带骨架的可视化,结果 JSON 存进predictions/。想换更高精度或更小体积的模型,只改--pose2d别名即可。
数据流水线与模型模块怎么工作
数据流水线在 mmpose/datasets/,负责把 COCO、MPII 等 30+ 种数据集统一成同一种格式,并在线做旋转、翻转、缩放等增强。
模型本体在 mmpose/models/,按 backbone 提特征、head 出关键点或热图、loss 算误差三段拆开,可以任意组合。换骨干网络或输出头时,其余部分基本不动。
一条典型配置就是在model下声明 backbone、head、loss 三段,改输入分辨率和关键点数量即可切换任务,不需要重写下层代码。
真实落地场景
健身动作评分(消费端):输入一段跟练视频,MMPose 逐帧提取全身 133 关键点,再和标准动作比对关节角度,输出每帧的相似度分。projects/just_dance/ 就是这个思路的完整示例。
多人密集场景质检(工业/研究):输入监控画面,bottom-up 模型一次标出所有人的关键点,配合 NMS 和实例合并得到每个人的骨架,用于异常行为或合规检测。
| 模型 | 输入 | AP (COCO-WholeBody) | 定位 |
|---|---|---|---|
| RTMW-S | 256×192 | 67.8 | 约 8MB,移动端 |
| RTMW-M | 256×192 | 70.1 | 实时推理 |
| RTMW-L | 256×192 | 72.3 | 高精度 |
容易踩的坑与调优思路
显存不够?先把 batch size 降下来,或换更小的 RTMPose-M/S;开 FP16 混合精度能再省一截。
关键点飘、遮挡处标错?检查裁剪的人框是否太松;单人 top-down 精度高,密集多人请切 bottom-up(RTMO / YOLOX-Pose)。
精度和速度要平衡?在 configs 里换 backbone 和输入分辨率,同一套 head 基本不用改,跑 tools/analysis_tools/get_flops.py 对比吞吐再定。
周边生态
姿态估计通常要配人检和跟踪:人框来自 MMDetection,跨帧跟踪交给 MMTracking,部署走 MMDeploy。
projects/ 目录下还有 PoseAnything、RTMPose3D 等现成扩展,可直接 fork 改。
打开 demo/ 目录,先跑一遍inferencer_demo.py看骨架效果。
【免费下载链接】mmposeOpenMMLab Pose Estimation Toolbox and Benchmark.项目地址: https://gitcode.com/GitHub_Trending/mm/mmpose
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考