我是tf的新手,我正在建立一个图像分类器。
问题是即使输入图像发生变化,使用函数sess.run()
的预测概率也不会改变。
我已经构建了模型,并在下面给出的代码中,我正在加载它并尝试预测图像。
def predict():
train = data.train
tf.reset_default_graph()
with tf.Session() as sess:
new_saver = tf.train.import_meta_graph('~/trained-model.ckpt.meta')
new_saver.restore(sess, '~/trained-model.ckpt')
print(tf.get_default_graph().get_name_scope())
#contains the softmax output of last_layer
y_pred = tf.get_default_graph().get_tensor_by_name('y_pred:0')
X = tf.get_default_graph().get_tensor_by_name('X:0')
final = imageprepare('tshirts.png')
final = np.asarray(final)
final = np.reshape(final,[784,1]
output_label = sess.run(y_pred, feed_dict={X: final})
print(output_label)
即使我更改了输入图像,output_label
也没有改变(tshirts.png)
整个代码为https://codeshare.io/2jO4oR我们将不胜感激任何帮助。提前谢谢!