动作理解框架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官方文档

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值