MMpose初体验--多人姿态检测关键点检测
mmpose是商汤开源的一款基于 PyTorch 的姿态分析的开源工具箱,是 OpenMMLab 项目的成员之一。 主分支代码目前支持 PyTorch 1.5 以上的版本。
MMPose开源代码:https://github.com/open-mmlab/mmpose
一、安装环境:
官方教程:依赖环境 — MMPose 0.28.1 文档
安装过程中的问题:
Traceback (most recent call last):
File "demo/top_down_video_demo_with_mmdet.py", line 193, in <module>
main()
File "demo/top_down_video_demo_with_mmdet.py", line 81, in main
assert has_mmdet, 'Please install mmdet to run the demo.'
AssertionError: Please install mmdet to run the demo.
问题解决:mmcv-full的版本不对。
pip3 install mmdet
pip install -U openmim
pip uninstall mmcv-full
mim install mmcv-full
二、图片人体姿态检测:
python demo/bottom_up_img_demo.py associative_embedding_hrnet_w32_coco_512x512.py hrnet_w32_coco_512x512-bcb8c247_20200816.pth --img-path tests/data/coco/ --out-img-root vis_results
三、视频人体姿态检测:
python demo/top_down_video_demo_with_mmdet.py demo/mmdetection_cfg/faster_rcnn_r50_fpn_coco.py https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth configs/body/2d_kpt_sview_rgb_vid/posewarper/posetrack18/hrnet_w48_posetrack18_384x288_posewarper_stage2.py https://download.openmmlab.com/mmpose/top_down/posewarper/hrnet_w48_posetrack18_384x288_posewarper_stage2-4abf88db_20211130.pth --video-path data/TongjiDancerClub.MOV --out-video-root vis_results --use-multi-frames --online
总结:
检测效果不错,多人检测,就是速度太慢,1080跑视频检测的时候都卡不能动。