我使用Tensorflow MobileNet retrain.py 示例文件(在https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py中)训练了分类网

当我使用经过训练的网络时,我设法一次只运行一个图像(输入是张量流4d阵列形状[1,128,128,3])

不知道这是否意味着什么,但列车过程中使用了火车批量标志 --train_batch_size=100

当我尝试对一批图像进行分类时(例如,tensorflow 4d数组形状[2,128,128,3]到 'input' 层),我得到以下错误: ValueError: Cannot feed value of shape (2, 128, 128, 3) for Tensor 'input:0', which has shape '(1, 128, 128, 3)'

(在运行此受过训练的网络会话之前,我正在使用预处理张量流会话,为此网络准备图像(调整大小,规范化等)

有谁知道我应该怎么做才能在这样的网络上运行一批图像,或者我如何配置 retrain.py 文件,这将创建一个允许批量运行的网络?