安装:
具体参考https://github.com/facebookresearch/SlowFast/blob/master/INSTALL.md
注意点:
1.使用conda install pytorch torchvision cudatoolkit=9.2
在自己服务器的虚拟环境下安装pytorch时需要使得cudatoolkit与驱动版本(使用nvidia-smi
查看)相匹配,不然torch.cuda.is_available()
返回false
使得安装的pytorch无法使用gpu加速;版本配对如下
具体问题参考:https://blog.csdn.net/weixin_42388228/article/details/107375344
2.由于服务器有加密系统不能使用git clone,detectron2,fvcore我均是在github上下载了zip文件然后解压再python setup.up install
数据准备:
具体参考https://github.com/facebookresearch/SlowFast/blob/master/slowfast/datasets/DATASET.md
这里只说config文件TRAIN.DATASET为kinetics,也就是上面网址的第一种情况。如果只训练的话得准备train.csv和val.csv,如果还要test的话还需要准备一个test.csv,这些文件放置的目录在DATA.PATH_TO_DATA_DIR path_to_your_dataset(也就是说DATA.PATH_TO_DATA_DIR后跟的目录中只含有csv文件)
每个csv的文件如下所示
path_to_video_1 label_1
path_to_video_2 label_2
path_to_video_3 label_3
...
path_to_video_N label_N