使用 Tensor Flow CNN 进行预测

Question

我是张量流的新手。 我设法训练和验证了 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

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