1

すでに3チャンネルのカラー画像で訓練された、あらかじめ訓練されたモデルを使用しようとしましたが、形状差のためにエラーが発生しています。誰かが私にこの問題にどのように取り組むことができるか教えてもらえますか? 1人のユーザーがTileレイヤーの使用を提案しましたが、このレイヤーまたは他のソリューションを使用するための関連ドキュメント/ヘルプが見つかりませんでした。私は本当にあなたの助けに感謝します。3チャンネル画像のあらかじめ訓練されたモデルを1チャンネル画像に適用するにはどうすればよいですか?

+0

モデルを再トレーニングしたくない場合は、同じチャネルデータを3つすべてに送信できます。 – lnman

+0

あなたはもっと詳しく説明できますか?私はその地域の専門家ではない。 –

+2

グレースケールイメージがあると仮定しています。 rgbチャネルのグレースケール値を繰り返すことで、それをrgbイメージに変換することができます。これを行うには 'rgb = cv2.cvtColor(gray、cv2.COLOR_GRAY2RGB)'のようなopencv関数があります。その後、その画像をあなたの事前に訓練されたモデルに使用することができます。 – lnman

答えて

2

タイルレイヤーについては、caffe.protoにはあまり情報がありません。コードを見ると、外の次元ごとにデータtilesがコピーされます。あなたの場合、それはそうでなければなりません:

layer{ 
    name: "tile" 
    type: "Tile" 
    bottom: "bottom-blob" 
    top: "top-blob" 
    tiling_param { 
     // axis is 1 by default 
     tiles: 3 
    } 
} 
関連する問題