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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章