在测试flownet2.0单张图片时,输入
run-flownet.py FlowNet2/FlowNet2_weights.caffemodel.h5 FlowNet2/FlowNet2_deploy.prototxt.template 0000000-img0.ppm 0000000-img1.ppm 0000000-pre.flo
第一次报错
Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM
解决办法
1、google到的方法是将cudnn从5.0升级到6.0,我没尝试;
2、将FlowNet2_deploy.prototxt.template中convolution_param 中加上“engine:CAFFE”;
3、在Makefile.config文件中将use_cudnn=1关掉,重新编译caffe;
尝试了第三种方法后
`
第二次报错
于是利用vim将将FlowNet2_deploy.prototxt.template中convolution_param 中加上“engine:CUDNN”全部替换为“engine:CAFFE”;
就成功了: