安装的各种包的版本:
- #python 3.9(MMAction要求python版本3.7+)
- #pytorch 2.0.1(MMAction要求pytorch版本1.8+)
- #torchvision 0.15.2
- #cuda 11.7(MMAction要求cuda版本10.2+)
- #mmaction 1.1.0
- #mmcv 2.0.0
预先准备工作:
- 安装miniconda,详情参考一下链接:miniconda3的安装及使用教程,此处不做赘述。
- 创建虚拟环境并激活,注意MMAction支持3.8以上版本的python。
conda create -n mmact python=3.9 conda activate mmact
-
安装pytorch,详情参考链接pytorch的安装,注意此处我安装的是gpu的版本。如果电脑没有英伟达显卡,请安装cpu版本。
1.安装mmcv(注意mmcv与pytorch、cuda版本兼容问题)
- 方法一:直接命令行安装,详情参考:pytorch、cuda对应mmcv版本及mmcv的安装
- 方法二:手动安装,需要将mmcv下载到本地,再手动pip安装,详情参考mmcv手动安装
2.安装MIM
- 方法一:(此方法的弊端比较明显,常因为网络问题出现安装失败的情况,建议使用方法二安装)
pip install git+https://github.com/open-mmlab/min.git
- 方法二: 手动安装
1)如图,直接去github把项目download下来
2)将下载下来的压缩文件解压,并找到其中setup.py文件所在路径
3)命令行cd进入setup.py所在目录,然后输入以下命令:
python setup.py install
此处安装会error提示缺少依赖,缺少什么依赖就直接命令行“pip install 包名==版本号 ”就行。
4)如图,安装成功
5)mmaction2安装参考以上步骤。
3.使用MIM安装MMEngine、MMDection(可选)、MMPose(可选)和MMAction2
min install mmengine
min install mmdet
min install mmpose
4.安装MMAction
- 方法一:从源码构建MMAction2
1)克隆MMAction2 repository.
conda install git
git clone https://github.com/open-mmlab/mmaction2.git
cd mmaction2
2)安装build requirements 以及安装MMAction2.
pip install -r requirements/build.txt
pip install -v -e .
- 方法二:手动安装(参考2方法二)
5.验证是否安装成功
在python解释器下运行如下代码:
import torch
from mmaction.apis import init_recognizer, inference_recognizer
config_file = 'configs/recognition/tsn/tsn_r50_video_inference_1x1x3_100e_kinetics400_rgb.py'
device = 'cuda:0' # 或 'cpu'
device = torch.device(device)
model = init_recognizer(config_file, device=device)
inference_recognizer(model, 'demo/demo.mp4')
6.测试结果
[(270, 31.022188), (230, 27.026155), (117, 24.818554), (318, 24.535355), (282, 24.215961)]