Tensorflow训练和测试时输入数据数量不同的解决方法

分为以下两种方法： 1 将训练的模型保存为ckpt格式，训练时一次训练batch_size张图片（比如设为8，16，32等），测试时再重新设置不同的batch_size(比如一次只想测试一张图片，则将batch_size设为1)，然后进行inference并restore保存的ckpt模型 2 如果将模型保存为pb格式，那么模型被固化，测试时数据的shape必须与训练时输入的数据shape保持一样

>>阅读原文<<