openmmlab实战营第二期--mmdetection和mmpose安装

安装 Pytorch

!pip3 install install torch1.10.1+cu113 torchvision0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

​

用MIM安装MMCV

!pip install -U openmim

!mim install mmengine

!mim install ‘mmcv==2.0.0rc3’

!mim install “mmdet>=3.0.0rc6”

安装其它工具包

!pip install opencv-python pillow matplotlib seaborn tqdm pycocotools -i https://pypi.tuna.tsinghua.edu.cn/simple

​

下载 MMPose

删掉原有的 mmpose 文件夹（如有）

!rm -rf mmpose

从 github 上下载最新的 mmpose 源代码

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

进入主目录

import os

os.chdir(‘mmpose’)

安装 MMPose

!mim install -e .

下载预训练模型权重文件和视频素材

import os

​

创建 checkpoint 文件夹，用于存放预训练模型权重文件

os.mkdir(‘checkpoint’)

​

创建 outputs 文件夹，用于存放预测结果

os.mkdir(‘outputs’)

​

创建 data 文件夹，用于存放图片和视频素材

os.mkdir(‘data’)

​

os.mkdir(‘data/test’)

多人图片，来源：https://www.pexels.com/zh-cn/photo/2168292/

!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220610-mmpose/images/multi-person.jpeg -O data/test/multi-person.jpeg

​

单人视频-唱跳篮球

!wget https://zihao-openmmlab.obs.myhuaweicloud.com/20220610-mmpose/videos/cxk.mp4 -P data/test

​

妈妈和女儿跳舞，经微信压缩

!wget https://zihao-openmmlab.obs.myhuaweicloud.com/20220610-mmpose/videos/mother_wx.mp4 -P data/test

​

两个女生跳舞视频，来源：https://mixkit.co/free-stock-video/two-girls-having-fun-in-a-retro-restaurant-42298/

!wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220610-mmpose/videos/two-girls.mp4 -O data/test/two-girls.mp4

检查安装成功

检查 Pytorch

import torch, torchvision

print(‘Pytorch 版本’, torch.version)

print(‘CUDA 是否可用’,torch.cuda.is_available())

检查 mmcv

import mmcv

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())

检查 mmpose

import mmpose

print(‘mmpose版本’, mmpose.version)

没有报错，即证明安装成功。