mmpose从安装到运行(必成功!!!!)

大模型研究僧

于 2025-03-12 16:59:33 发布

阅读量572

点赞数 5

文章标签：深度学习人工智能姿态估计

本文链接：https://blog.csdn.net/weixin_44006686/article/details/146203212

版权

MMpose作为一个姿态估计的库，用的人比较广泛，因此本文章整理了mmpose从安装到运行到配置的全过程。

笔者装了一天这个，该踩得坑全踩了，哎。祝大家一切顺利！

特别感谢：

低版本安装v0.x

40系显卡请跳过低版本!!!

创建新环境

conda create -n openmmlab python=3.8

检查cuda是否可用

在这之前,你需要安装配置好cuda和cudnn
这里不提供教程，需要你安装cuda（CSDN有很多非常详细非常好的教程，完成这一步之后再来）

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c pytorch

安装到这里之后可以看看的cuda是否可用如果可用跳过下面这一步骤

pip install torch==1.10.0+cu102 torchvision==0.11.0+cu102 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

判断GPU可用？

import torch
print(torch.__version__)  # PyTorch版本
print(torch.cuda.is_available())  # CUDA是否可用

在python终端中执行该代码即可
这一步返回应该是True 如果为false说明你的cuda和torch不兼容需要重新配置.(去找教程)
在这里插入图片描述

安装mmcv

激活你的conda环境

安装MVCC-FULL
这里需要注意的是你安装的那个cuda版本和torch版本如果前面是按照我的步骤来的不需要改

pip install mmcv-full==1.5.3 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html

git下载mmpose包

这里我不建议使用最新的mmpose(使用git clone的方式),最好手动打开然后下载下来,最新版本的兼容性实在是太差了
进去手动下载: link

安装依赖包

cd mmpose
pip install -r requirements.txt
pip install -v -e .

安装mmdet

pip install mmdet==2.25.1
pip install -U openmim

验证结果

运行demo

python demo/top_down_img_demo.py configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192.py  https://download.openmmlab.com/mmpose/top_down/hrnet/hrnet_w48_coco_256x192-b9e0b3ab_20200708.pth  --img-root tests/data/coco/ --json-file tests/data/coco/test_coco.json --out-img-root vis_results

高版本安装v1.x

注意:40系显卡必须这么安装!

安装cuda和torch

安装cuda的11.8工具

conda install cudatoolkit=11.8

安装torch

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118

安装mmpose以及相关包

pip install -U openmim

如果pip报错则尝试降级,没有就跳过:

python -m pip install pip==20.0.2

安装相关依赖

mim install mmengine
mim install "mmcv>=2.0.1"
mim install "mmdet>=3.1.0"

这里是最可能出现各种冲突的,而且比较难解决
在这里插入图片描述
去官网(必须去别浪费时间!!!)

在这里找到你对应的版本然后安装就好了

pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0/index.html

安装mmpose

从git下载下来

git clone https://github.com/open-mmlab/mmpose.git

cd mmpose
pip install -r requirements.txt
pip install -v -e .

测试

mim download mmpose --config td-hm_hrnet-w48_8xb32-210e_coco-256x192  --dest

python demo/image_demo.py  tests/data/coco/000000000785.jpg    td-hm_hrnet-w48_8xb32-210e_coco-256x192.py  td-hm_hrnet-w48_8xb32-210e_coco-256x192-0e67c616_20220913.pth   --out-file vis_results.jpg --draw-heatmap

在这里插入图片描述
都到这里了，希望大家都能顺利安装。