st-gcn配置环境的过程中可能会遇到的问题

1、关于caffe和opencv的安装可以尝试以下安装方式:
openpose在cmake安装过程中,遇到无法生成编译文件的情况,(暂时未解决)
在conda环境,在先安装opencv情况下,再安装caffe-gpu,会出现提示 移除opencv的命令提示,不清楚具体原因。
[安装caffe]
gpu版:conda install -c defaults caffe-gpu
cpu版:conda install -c defaults caffe
[安装opencv]
conda install --channel https://conda.anaconda.org/menpo opencv3
2、作者给出的st-gcn测试
测试的具体流程可以按照github上的:

python main.py recognition -c config/st_gcn//train.yaml [–work_dir ] –device 0 --batch_size 128

其中dataset必须为:ntu-xsub, ntu-xview, kinetics-skeleton之一;
work_dir 是训练结果保存的位置,默认为work_dir, 也可以自动设置;
如果显存不够,则减少batch_size;
3、自己进行训练
(1)github上给出了如下的具体的训练方法,如果速度上不来,可以更改batch_size的大小,作者也给出了两个参考值:
在./config/st_gcn/文件夹中有三个不同的子文件夹:kinetics-skeleton, ntu-xsub, ntu-xview,每个文件夹下面都有两个文件:train.yaml 和 test.yam

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值