TSM复现以及出现的错误

- 论文与源码
论文:https://arxiv.org/pdf/1811.08383.pdf
github地址:https://github.com/mit-han-lab/temporal-shift-module
平台:极链AI云
- 环境:
在这里插入图片描述

# 创建虚拟环境
conda create -n tsm python=3.8
conda activate tsm

# 安装cuda11.1的pytorch1.8和torchvision0.9.0
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html


pip install TensorboardX
pip install  tqdm
pip install scikit-learn

- 数据准备:
UCF101的前三类,因为我之前拿mmaction处理过就不多说了,主要步骤为:将视频处理成帧,利用标签文件,官方划分训练集和测试集的txt文件,最后得到训练集与与验证集的划分文件
可以看这个博主的文章:行为识别TSM训练ucf101数据集

-

评论 19
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值