使用TensorFlow卷积神经网络进行图像分类

Question

我是一个有用的助手，可以翻译文本。

我对使用CNN进行图像分类非常陌生，并且我遵循了谷歌CNN的教程：https://www.tensorflow.org/tutorials/images/cnn 这个教程很好，但是在训练和评估模型之后突然停止了。我不知道为什么会停止 :(

是否有人知道是否有类似的Python函数：

loadModel("/path/to/my/trained/model");    
classify("/imagepath/image.png");

那么返回给我找到的对象的类别和所有像素，如果可能的话。或者有什么方法可以做到这一点吗？

- Cracker0dks

---

这个问题将会帮助你解决同样的问题。 - turtle

我在你提供的链接中找不到任何可能的解决方案。你能具体一点吗？ - Cracker0dks

Answer 1

首先，您需要更具体地说明“停止工作”的问题。出了什么问题？您是否遇到错误或在某个地方卡住了？ 其次，this 可能对您有用。

- Elmira

---

不，我的意思是教程停了。没有更多关于如何使用训练好的网络的信息了。 - Cracker0dks

Answer 2

#to save a model
sess = tf.Session()
saver.save(sess, '/path/to/my/saved/model/my_test_model)

在TensorFlow中加载已保存的模型

with tf.Session() as sess:    
    saver = 
    tf.train.import_meta_graph('/path/to/my/saved/model/my_test_model-
    1000.meta',clear_devices=True)
    saver.restore(sess,tf.train.latest_checkpoint('./'))

在已加载的模型上进行预测

sess.run(tf.argmax(y, 1), feed_dict={x: [img]})

- thefifthjack005