使用TensorFlow CNN进行图像分类

Question

我对使用CNN的图像分类非常陌生，我遵循谷歌CNN的教程：tensorflow.org/deepCnn

这种方法效果非常好，但在训练和评估模型后突然停止。不知道他们为什么停止:(

有没有人知道是否有像以下一样的python函数：

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

返回我的类，如果可能的话，返回找到的对象的所有像素。或者说是这样做的？

谢谢：）

Answer 1

首先，您需要更加具体地说“停止工作”。问题是什么？你得到错误还是冻结在哪里？ 其次，this对您有用

Answer 2

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

在张量流中加载已保存的模型

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]})