MMLab系列的环境一定要对应好版本以及官方文档,不让会出现一系列很神奇的红色错误。
我这里用的是趋动云,因为有活动,基本白嫖
趋动云里面内置的有cuda和pytorch环境,直接选择即可,不过有时候加载是真的慢。
下面进入正题,MMPose的安装,我这里的环境是pytorch1.10+cuda11.2
gcc version 9.3.0 (Ubuntu 9.3.0-17ubuntu1~20.04)
我是用vscode连接的这个云平台**(强烈推荐)**
pip install -U openmim
mim install mmengine
mim install 'mmcv==2.0.0rc3'
mim install "mmdet>=3.0.0rc6"
下面这个红色可能是因为超时,因此我的解决方案是加上清华源
果然。。。。。。。。。。
他就正常了
然后就是下载其他的用得上的东西了:
ip install opencv-python pillow matplotlib seaborn tqdm pycocotools -i https://pypi.tuna.tsinghua.edu.cn/simple
接着
git clone https://github.com/open-mmlab/mmpose.git -b tutorial2023
注意分支是tutorial2023 ,可能其他的分支版本上有些问题
这个代码拉下来之后,进入到mmpose里面,编译安装它
进去之后一定要执行这一步!!!!!!!!!!
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
不然编译下载找不到文件。
最后通过
mim install -e .
安装即可
接着就是测试了!!!
python demo/image_demo.py \
tests/data/coco/000000000785.jpg \
td-hm_hrnet-w48_8xb32-210e_coco-256x192.py \
hrnet_w48_coco_256x192-b9e0b3ab_20200708.pth \
--out-file vis_results.jpg \
--draw-heatmap```
![请添加图片描述](https://img-blog.csdnimg.cn/589d2eb7a71f49edbb77577b463ee347.png)