Caffeレコードエラー
567 ワード
：source param shape is 20 3 5 5 ;target param shape is 20 1 5 5 caffeのmnistネットワークで自分のデータを訓練して得られたcaffemodelは、実行するたびに異なる結果を得る.その後予測時に自分で生成するdeploy.prototxtも同様にmnistのlentに置き換える.prototxt、このエラーが発生しました.原因:元のshapeが一致せず,cov 1層で発生した誤りであることが明確に指摘された.だから直接shapecpnv 1を探して、それからやっと訓練モデルファイルがモデル定義ファイルのshapeと一致しないことを発見して、私が訓練する時画像ライブラリの中のある画像は1つのチャネルで、あるのは使用する3つのチャネルで、だからデフォルトは3つのチャネルを使って、私のモデル定義ファイルの中のshape、そのチャネル数は1つのチャネルで書いて、だから間違いが発生します.minstが持参したピクチャセットは白黒2色で、画素値ごとに1つのチャネルしかありませんが、私のピクチャも白黒ですが、実はRGB形式です.解決策はlent.prototxtにおけるInput層の2番目のdimを3に変更する.