使用Tensor Flow CNN进行预测

Question

我是tensorflow的新手。我已经训练并验证了CNN，通过Saver对象将会话保存到CPKT文件中并加载回来。现在我想使用训练有素的模型来检查它对我拍摄的照片的表现。我无法弄明白该怎么做。

Answer 1

• 使用tf.train.Saver()对象

  saver = tf.train.Saver()
  with tf.Session() as sess:
      saver.restore(sess, tf.train.latest_checkpoint('./'))
  

Answer 2

很高兴听到！您需要做的只是load the checkpoint，并根据它评估网络。这通常使用与训练脚本相同的脚本完成，但仅评估批量大小为1的损失函数。