caffe編譯/執行錯誤
阿新 • • 發佈：2019-01-30
作者：xg123321123
宣告：版權所有，轉載請聯絡作者並註明出處
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；以此類推。