Tensorflow: модель не загружается - PullRequest
0 голосов
/ 26 мая 2018

Я новичок в 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 не меняется, даже если я изменяю входное изображение (футболки.png)

Весь код https://codeshare.io/2jO4oR Любая помощь будет оценена.Заранее спасибо!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...