本文将引导快速使用 MMSkeleton ,介绍用摄像头测试实时姿态估计。
- MMSkeleton: https://github.com/open-mmlab/mmskeleton
安装
首先安装 MMDetection ,可见 MMDetection 使用。
然后安装 MMSkeleton ,
# 启用 Python 虚拟环境
conda activate open-mmlab
# 下载 MMSkeleton
git clone https://github.com/open-mmlab/mmskeleton.git
cd mmskeleton
# 安装 MMSkeleton
python setup.py develop
# 安装 nms op for person estimation
cd mmskeleton/ops/nms/
python setup_linux.py develop
cd ../../../
现有模型,视频测试
配置
configs/pose_estimation/pose_demo.yaml
:
processor_cfg:
video_file: resource/data_example/ta_chi.mp4
detection_cfg:
model_cfg: ../mmdetection/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.py
checkpoint_file: ../mmdetection/checkpoints/cascade_rcnn_r50_fpn_1x_coco_20200316-3dc56deb.pth
bbox_thre: 0.8
选用的检测模型,如下:
- Cascade R-CNN, R-50-FPN, 1x