Tensorflow訓練和測試時輸入數據數量不同的解決方法

分爲以下兩種方法： 1 將訓練的模型保存爲ckpt格式，訓練時一次訓練batch_size張圖片（比如設爲8，16，32等），測試時再重新設置不同的batch_size(比如一次只想測試一張圖片，則將batch_size設爲1)，然後進行inference並restore保存的ckpt模型 2 如果將模型保存爲pb格式，那麼模型被固化，測試時數據的shape必須與訓練時輸入的數據shape保持一樣

>>阅读原文<<