1

私はVGG_ILSVRC_19_layersを事前に訓練されたモデルとして、数字の異なるデータセットで使用したいと考えています。 別のラベルファイルが必要ですか?このモデルをアップロードしてデータセットに使用するにはどうすればよいですか?私はレイヤーを変更する方法DIGITS(別のラベル)で異なるデータセットの新しい事前訓練モデルを使用するにはどうすればいいですか?

ERROR: Cannot copy param 0 weights from layer 'fc6'; shape mismatch. Source param shape is 1 1 4096 25088 (102760448); target param shape is 4096 32768 (134217728). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.

を得たVGG 16層のための

答えて

1

あなたのラベルは、あなたのモデルではなく、DIGITSのデータセットに関連付けられています。あらかじめ訓練されたVGGモデルをアップロードするときは、モデルが1000ではなくNクラスで動作するように、最後の内部製品レイヤー(this answerを参照)の名前を変更する必要があります。

+0

回答ありがとうございます。あなたの配備ネットワークにはSoftmaxレイヤーがありません!カスタムネットワークのドキュメントを読んだり、例については標準ネットワークを見てください。 " –

+0

これは別の問題です。 SOやGitHubに別の質問を投稿しても構いません。 –

+0

ああ、あなたはすでにやった:http://stackoverflow.com/q/40143901/2404152 –

関連する問題