OpenMMLab 训练营第二期 笔记三

RTMPose 关键点检测

环境配置

我的电脑是 Windows 系统，安装了 miniconda。
不同项目需要不同的版本，使用 conda 更容易管理。

首先来安装创建 RTMPose 的环境¹。

# 以下命令都在 minicoda shell 执行
# 查看所有环境
conda info -e

# 创建 RTMPose 环境
conda create -n RTMPose python=3.8

# 切换到 RTMPose 环境
conda activate RTMPose

# 安装Pytorch
pip3 install install torch==1.10.1+cu113 torchvision==0.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.0"
mim install "mmdet>=3.0.0rc6"

# 安装其它工具包
pip install opencv-python pillow matplotlib seaborn tqdm pycocotools -i https://pypi.tuna.tsinghua.edu.cn/simple

# 从 github 上下载最新的 mmpose 源代码
git clone https://github.com/open-mmlab/mmpose.git -b tutorial2023

# 进入主目录
cd mmpose

# 安装 MMPose 
mim install -e .

# <begin> 下载预训练模型权重文件和视频素材 -----------