Tensorflow - チュートリアルで説明したようにCNNモデルを構築する

Question

TFレイヤーガイドの実装を完了しました.NMISTデータセットの畳み込みニューラルネットワークを構築しました。トレーニングモデルは正常に実行され、97.3％の精度を与えました。

しかし、チュートリアルでは、この新しい訓練されたモデルを使用して独自の画像を提供し、予測を表示する方法については言及していません。誰もがトレーニングモデルの出力を使って予測を行う方法を知っていますか？ tmp/mnist_convnet_model $フォルダには、.pbtxt、メタファイル、インデックスファイルなどの出力ファイルがいくつかあります。しかし、自分のイメージを予測するための指示を見つけることができません。

Answer 1

y_pred = tf.nn.softmax(your_final_layer) 
y_pred_cls = tf.argmax(y_pred, dimension=1) 

と予測

feed_dict = {x: [your_image]} 
classification = tf.run(y_pred_cls, feed_dict) 
print classification 

のためにこれはただのモデルについて、あなたは

を作成するために適用されます