SlowFast 运行Charades数据集
平台:Ubuntu16.04
项目地址: https://github.com/facebookresearch/SlowFast
友情提示:torch版本尽量高于等于1.5
第一步:
按照install.md进行环境配置。
你可能在安装detectron2的时候出现问题,但是如果你是Ubuntu就不要慌。我们来到detectron2安装的install说明中,找到下图。
按照你的torch版本点击相应的install按钮即可,把相应语句运行即可。当然你也可以按照这个教程进行问题解决,但我的g++问题还是没有解决。
对了,你只需完成到build pySlowFast那一步即可,最后一步可以不用,因为你的数据集还未准备。
然后点击data preparation这个说明当中。
找到Charades数据集那一项,下载相应文件即可。注意不需要下载多余的标记文件,因为detectron有目标检测的功能。路径记得用引号变为字符。
然后下载pretrain好的模型参数,https://github.com/facebookresearch/SlowFast/blob/master/MODEL_ZOO.md。找到Charades数据集的位置,如下。
根据你想用
configs/Charades
下哪一个配置文件选择下载。我使用的没有multigrid的.yaml文件,所以下载的是第一行的pretrain文件。将下载好的pretrain文件放在checkpoint文件下。
接着就是修改配置文件。
运行
python tools/run_net.py --cfg configs/Charades/SLOWFAST_16x8_R50.yaml
如果没有出错,那么就可以成功运行。
如果报错
AssertionError(2, 4)
你安装报错的改一下即可,把配置文件里相应的4改成2.