caffe_faster_rcnn_踩坑

最新推荐文章于 2024-09-14 17:02:52 发布

weixin_30682415

最新推荐文章于 2024-09-14 17:02:52 发布

阅读量70

点赞数

文章标签：人工智能 python

原文链接：http://www.cnblogs.com/chinesezyc/p/6951942.html

版权

一. 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/

转载于:https://www.cnblogs.com/chinesezyc/p/6951942.html