Tensorflow:模型未加载

时间:2018-05-26 19:43:52

标签: tensorflow machine-learning image-recognition

我是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我们将不胜感激任何帮助。提前谢谢!

0 个答案:

没有答案