RC3D时序动作检测代码调试工作
环境安装
git地址:https://github.com/sunnyxiaohu/R-C3D.pytorch
ubuntu18.04+CUDA9.0: 环境安装教程
anaconda环境如下:
工程编译
1.需要生成训练文件:
下载 thumos14/activity1.3数据集,在preprocess/thumos14文件夹下:
第一步将训练集处理成帧图像,第二步生成训练文件。
2.编译:
cd lib
sh make.sh
需要注意的是:
需要修改make.sh文件,更改CUDA_ARCH和自己的GPU匹配
需要添加一些编译选项,如下图:
测试运行
./script_train.sh res18 activitynet --gpus 0 --nw 4 --lr 0.0001 --lr_decay_step 6