首页 文章

使用Tensor Flow CNN进行预测

提问于
浏览
1

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

2 回答

  • 1
    • 使用 tf.train.Saver() 对象
    saver = tf.train.Saver()
    with tf.Session() as sess:
        saver.restore(sess, tf.train.latest_checkpoint('./'))
    
  • 0

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

相关问题