caffe_faster_rcnn_踩坑

一. python3.5 修改ROOTDIR/setup.py中的dict属性为items()

 

二.cudnn error

/include/caffe/util/cudnn.hpp: In function ‘void caffe::cudnn::createPoolingDesc(cudnnPoolingStruct**, caffe::PoolingParameter_PoolMethod, cudnnPoolingMode_t*, int, int, int, int, int, int)’:
./include/caffe/util/cudnn.hpp:127:41: error: too few arguments to function ‘cudnnStatus_t cudnnSetPooling2dDescriptor(cudnnPoolingDescriptor_t, cudnnPoolingMode_t, cudnnNanPropagation_t, int, int, int, int, int, int)’

解决,因为rgb用的cudnnV4,我们要替换成V5的

cp /home/zyc/deep_obj/caffe-master/include/caffe/layers/cudnn_* caffe-fast-rcnn/include/caffe/layers/
cp /home/zyc/deep_obj/caffe-master/src/caffe/layers/cudnn_* caffe-fast-rcnn/src/caffe/layers/
cp /home/zyc/deep_obj/caffe-master/include/caffe/util/cudnn.hpp caffe-fast-rcnn/include/caffe/util/

 

posted @ 2017-06-06 15:46  O1dCat  阅读(189)  评论(0编辑  收藏  举报