Я новичок в 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 Любая помощь будет оценена.Заранее спасибо!