TSN现在差不多被SlowFast取代了,但这两天在公司还是在用它抽特征。直接装python-opencv还要额外配置GPU的支持,所以用bitxiong/tsn还是个不错的选择。
这里就列几个坑点,做完这些可以直接调用用GPU抽光流图片的脚本了,但是如果版本驱动对不上,模型还是跑不起来,可能就抽不了特征。但抽特征的代码用pytorch还是很容易写的,不必依赖这套caffe代码。
- enable GPUs with docker:docker run --gpus all
- 安装openmpi再build_all.sh
- GPU算力太高,不匹配
- cmd空格改等号,默认文件格式改MP4,路径去掉一层文件夹