windows安装mmaction2以及验证推理

MMAction2 is an open-source toolbox for video understanding based on PyTorch. It is a part of the OpenMMLab project.

Action Recognition on Kinetics-400

Skeleton-based Action Recognition on NTU-RGB+D-120

Skeleton-based Spatio-Temporal Action Detection and Action Recognition Results on Kinetics-400

Spatio-Temporal Action Detection Results on AVA-2.1

1.环境安装

我这里安装的是虚拟环境，建议大家使用anaconda虚拟环境，这样不同的模型的环境不会冲突，更方便管理。

conda create --name openmmlab python=3.8 -y
conda activate openmmlab
conda install pytorch torchvision -c pytorch  # This command will automatically install the latest version PyTorch and cudatoolkit, please check whether they match your environment.
pip install -U openmim
mim install mmengine
mim install mmcv
mim install mmdet  # optional
mim install mmpose  # optional
git clone https://github.com/open-mmlab/mmaction2.git
cd mmaction2
pip install -v -e .

1. 先创建好虚拟环境，再激活该环境

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

然后再安装pytorch,请注意pytorch的版本要和cuda版本匹配
可以使用nvcc -V 命令查看当前cuda版本，再去下面给出的pytorch官网找到符合自己的pytorch版本
https://pytorch.org/get-started/previous-versions/

我是cuda11.1，安装的是torch==1.8

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

2. 先安装openmim，然后使用它来安装mmcv、mmdet、mmpose，它会自动处理好它们之间的版本关系。mmdet和mmpose是和骨骼行为识别相关的包，如果不是研究骨骼行为识别的话，可以不安装mmdet和mmpose这两个包。

pip install -U openmim
mim install mmengine
mim install mmcv
mim install mmdet  # optional
mim install mmpose  # optional

3. 提前下载好mmaction2到本地，mmaction2下载地址
   我这里是放在D盘
   
4. 进入到下载好的mmaction2-main这个目录下
   
5. 最后执行pip install -v -e .mmaction2就安装好了
   

2.验证是否安装成功（推理验证）

python demo/demo_inferencer.py demo/demo.mp4 --rec tsn --print-result --label-file tools/data/kinetics/label_map_k400.txt

出现下图结果表示mmaction2已经安装成功了

使用测像头验证

python demo/webcam_demo.py demo/demo_configs/tsn_r50_1x1x8_video_infer.py checkpoints/tsn_r50_1x1x3_100e_kinetics400_rgb_20200614-e508be42.pth tools/data/kinetics/label_map_k400.txt --average-size 5 --threshold 0.2 --device cuda

3.验证posec3d（骨骼行为识别）

前面安装的mmdet，mmpose是骨骼行为识别需要的
使用下面的命令做posec3d的demo推理

python demo/demo_skeleton.py demo/demo_skeleton.mp4 demo/demo_skeleton_out.mp4

结果如下：

4.最后想说的话

这是在windows上安装mmaction2，要是想要在linux上安装的话，直接使用mmaction2在github上的安装教程就可以了，就是安装pytorch时要注意和cuda版本匹配。安装命令如下：

conda create --name openmmlab python=3.8 -y
conda activate openmmlab
conda install pytorch torchvision -c pytorch  # This command will automatically install the latest version PyTorch and cudatoolkit, please check whether they match your environment.
pip install -U openmim
mim install mmengine
mim install mmcv
mim install mmdet  # optional
mim install mmpose  # optional
git clone https://github.com/open-mmlab/mmaction2.git
cd mmaction2
pip install -v -e .

我是研究骨骼行为识别方向的研究生，如果也想研究骨骼行为识别的话这里推荐 pyskl 项目，它是是一个专门用于骨骼行为识别的开源工具箱，感兴趣的话可以看下我的专栏里的其它博客，介绍了pyskl的安装以及如何制作自己的骨骼行为数据集。