caffe编译/运行错误


作者:xg123321123

出处:http://blog.csdn.net/xg123321123/article/details/54094441

声明:版权所有,转载请联系作者并注明出处

1

报错:
fatal error: caffe/proto/caffe.pb.h: No such file or directory #include “caffe/proto/caffe.pb.h”

解决:
caffe根目录下运行

protoc src/caffe/proto/caffe.proto --cpp_out=.
mkdir include/caffe/proto
mv src/caffe/proto/caffe.pb.h include/caffe/proto

2

报错:
Check failed: status == CUBLAS STATUS_SUCCESS (11 vs. 0) CUBLAS STATUS MAPPING_ERROR

解决:
(1)caffe用OpenCV读取图像时,是读取的0–255,这样读取的label也是如此;
caffe如果读到了255,它就认为是有0–255的标签,但是我们最后的输出比这个维度小,这样就不对了,所以需要在数据读入层指定 scale
transform_param {
scale: 0.00390625
}

所以,制作标签时一定要注意从0开始一个个地往后排,不能弄错。

(2)如果分类类别数有2类,那么输入loss层的数据的channel应该为2;以此类推。


注意!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。



 
粤ICP备14056181号  © 2014-2019 ITdaan.com

赞助商广告