MMPose安装
安装
!rm -rf mmpose #rm -rf 删除目录
下载教程代码
git clone https://github.com/TommyZihao/MMPose_Tutorials.git
mv .\mmpose\MMPose_Tutorials ./MMPose_Tutorials #下错地址了
git clone https://github.com/open-mmlab/mmpose.git -b tutorial2023 # -b branch 下载分支
pip install -U openmim
mim install mmengine
mim install 'mmcv==2.0.0rc3'
mim install "mmdet>=3.0.0rc6"
cd ./mmpose
mim install -ve .
# pip install ipykernel
code .
pip 参数说明:
- v:verbose(冗长的), or more output. (输出更多的信息)
- e: installing a project in editable mode. (以可编辑的模式安装项目)
- . : 当前路径下
python的os.chdir():选择路径文件夹
os.mkdir():生成文件夹
案例目录说明:
- checkpoint 预训练模型权重
- outputs 存放预测结果
- data 存放图片和视频素材
- data/test
import mmpose
import mmcv
mmpose.__version__
from mmcv.ops import get_compiling_cuda_version, get_compiler_version
print('MMCV版本', mmcv.__version__)
print('CUDA版本', get_compiling_cuda_version())
print('编译器版本', get_compiler_version())
MMDetection安装
git clone https://github.com/open-mmlab/mmdetection.git -b 3.x
cd ./mmdetection
mim install -ve .
import mmdet
print('mmdetection版本', mmdet.__version__)
下载安装权重
cd ./mmdetection
mim download mmdet --config rtmdet-ins_tiny_8xb32-300e_coco --dest .
运行示例
python demo/image_demo.py demo/demo.jpg rtmdet-ins_tiny_8xb32-300e_coco.py --weights rtmdet-ins_tiny_8xb32-300e_coco_20221130_151727-ec670f7e.pth --device cpu