Convolutional Neural Network mit TensorFlow

Question

Ich baue ein CNN-Modell mit Tensorflow, ohne die Verwendung von Front-End-APIs wie Keras. Ich erstelle ein VGG-16-Modell und verwende die vortrainierten Gewichte und möchte die letzten Schichten fein abstimmen, um meinen Zweck zu erfüllen.

Nach dem Tutorial hier, http://cv-tricks.com/tensorflow-tutorial/training-convolutional-neural-network-for-image-classification/ Ich erstellte das Trainingsskript und modifizierte nach meinen Anforderungen. Mein Training findet jedoch nicht statt und die Trainingsgenauigkeit liegt bei 50,00% und die Validierungsgenauigkeit bildet ein Muster, das die Zahlen wiederholt. Attached ist der Screenshot des gleichen.

Ich bin seit Tagen auf diesem fest und kann nicht scheinen, den Fehler zu finden. Jede Hilfe wird geschätzt.

Der Code ist ziemlich lang und daher ist hier die gist file für die gleiche

Answer 1

Ihre Kreuzentropie falsch ist, werden Sie Ihre Logits mit dem softmax Ihrer Logits zu vergleichen. Diese

:

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

Sollte sein:

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

Einige Dinge zu beachten. Ich würde nicht an einem Datenpunkt trainieren und dann am selben Datenpunkt auswerten. Ihre Trainingsgenauigkeit wird wahrscheinlich dadurch verzerrt sein. Ein weiterer Punkt ist zu beachten, dass tf.argmax(tf.softmax(logits)) das gleiche wie ist.