TensorFlowを使用した畳み込みニューラルネットワーク

Question

KerasなどのフロントエンドAPIを使用せずに、Tensorflowを使用してCNNモデルを構築しています。私はVGG-16モデルを作成し、事前にトレーニングされたウェイトを使用しています。最後のレイヤーを自分の目的に合わせて微調整したいと思います。

ここでのチュートリアルの後にhttp://cv-tricks.com/tensorflow-tutorial/training-convolutional-neural-network-for-image-classification/ 私はトレーニングスクリプトを再作成し、自分の要件に従って修正しました。しかし、私の訓練は起こらず、訓練の正確さは50.00％で止まっており、妥当性確認の正確さは数字を繰り返すパターンを形成しています。 同じもののスクリーンショットが添付されています。

私は数日間この状態で立ち往生しており、エラーを見つけることができませんでした。どんな助けもありがとうございます。

コードはかなり長いですので、ここではgist file同じ

Answer 1

ためにあなたのクロスエントロピーが間違っていると、あなたはあなたのlogitsのソフトマックスを使用してlogitsを比較しています。

この：

cross_entropy = tf.nn.softmax_cross_entropy_with_logits(logits=layer_fc2, 
                labels=y_pred) 

は次のようになります。

cross_entropy = tf.nn.softmax_cross_entropy_with_logits(logits=layer_fc2, 
                labels=y_true) 

注意すべきいくつかの事。私はいくつかのデータ点について訓練をしないで、同じデータ点について評価しません。あなたのトレーニングの精度はおそらくそうすることによって偏っていくでしょう。もう一つの注意点は、tf.argmax(tf.softmax(logits))はtf.argmax(logits)と同じです。