动作理解框架mmaction2环境搭建
1. 创建 Python 虚拟环境
conda create -n mmaction2 python=3.8 -y
conda activate mmaction2
2. 安装 PyTorch 框架
这里我安装的是1.8.0版本的:pytorch官网
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
3. 使用 MIM 安装 MMEngine、MMCV、MMDetection(可选)和 MMPose(可选)
pip install -U openmim
mim install mmengine
mim install mmcv
mim install mmdet
mim install mmpose
4. 安装 MMAction2
采用的是官方推荐的从源代码构建 MMAction2的方式(方便后续在 MMAction2 框架上开发自己的动作识别任务或新功能。例如,添加新的数据集或新的模型。)
git clone https://github.com/open-mmlab/mmaction2.git
cd mmaction2
pip install -v -e .
#"-v" 表示输出更多安装相关的信息
#"-e" 表示以可编辑形式安装,这样可以在不重新安装的情况下,让本地修改直接生效。
5. 验证安装
为了验证 MMAction2 是否安装正确,官方提供了一些示例代码来运行推理演示。
第一步,下载配置文件和权重文件。
mim download mmaction2 --config tsn_imagenet-pretrained-r50_8xb32-1x1x8-100e_kinetics400-rgb --dest .
第二步,验证推理演示。
#demo.mp4 和 label_map_k400.txt 都来自于 Kinetics-400
python demo/demo.py tsn_imagenet-pretrained-r50_8xb32-1x1x8-100e_kinetics400-rgb.py tsn_imagenet-pretrained-r50_8xb32-1x1x8-100e_kinetics400-rgb_20220906-2692d16c.pth demo/demo.mp4 tools/data/kinetics/label_map_k400.txt
在终端看到前5个标签及其对应的分数。
可能出现
报错1:
ImportError: DLL load failed while importing _imaging: 找不到指定的模块。
参考DLL load failed while importing _imaging: 找不到指定的模块的解决方法,卸载当前pillow版本(我当前的pillow版本是9.3.0),安装6.2.1版本
pip uninstall pillow
pip install pillow==6.2.1 -i https://pypi.tuna.tsinghua.edu.cn/simple/
报错2:
AssertionError: MMCV==2.2.0 is used but incompatible. Please install mmcv>=2.0.0rc4, <2.2.0.
可重新安装mmcv:
mim uninstall mmcv
mim install mmcv==2.0.1
参考:
mmaction2官方文档