pyslowfast multigrid训练自己的数据集(没有把网络模型给抠出来再按照自己的训练,而是完全按照作者的代码命令训练)

安装
具体参考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
  • 4
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值