OpenMMLab AI实战训练营day3-MMpose代码教程

MMPose 是一款基于 Pytorch 的姿态估计开源工具箱，是 OpenMMLab 项目的成员之一，包含了丰富的 2D 多人姿态估计、2D 手部姿态估计、2D 人脸关键点检测、133关键点全身人体姿态估计、动物关键点检测、服饰关键点检测等算法以及相关的组件和模块，下面是它的整体框架：

MMPose 由 8 个主要部分组成，apis、structures、datasets、codecs、models、engine、evaluation 和 visualization。

环境部署

第 1 步 从官网 下载并安装 Miniconda。
第 2 步 创建一个 conda 虚拟环境并激活它。

conda create --name openmmlab python=3.8 -y
conda activate openmmlab

第 3 步 安装GPU版本pytorch和torchvison,个人建议下载whl文件安装，以下下给出网址
https://download.pytorch.org/whl/torch_stable.html

pip install *****.whi

第 4 步 使用 MIM 安装 MMEngine 和 MMCV

pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"

第 5 步 从源码安装mmpose

git clone https://github.com/open-mmlab/mmpose.git
cd mmpose
pip install -r requirements.txt
pip install -v -e .
# "-v" 表示输出更多安装相关的信息
# "-e" 表示以可编辑形式安装，这样可以在不重新安装的情况下，让本地修改直接生效

第 6 步 从源码安装mmdetection,操作格式同上

训练前准备

准备数据集，书写配置文件
这部分内容比较细节，推荐大家看同济子豪兄的讲解

训练

调用mmpose文件夹下tools/train.py 输入参数 开启训练