作者: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;以此類推。