尝试微调MobileNet_v1_1.0_224。
我面临的问题是输入:0有一个形状[1,224,224,3]。我想给它喂一批任意大小,例如[100,224,224,3]。我正在玩的代码是Tensorflow for poets,retrain.py是准确的。
以下是从输入打印的内容:0 tensor:
image_buffer_input = graph.get_tensor_by_name('input:0')
print (image_buffer_input)
> Tensor("input:0", shape=(1, 224, 224, 3), dtype=float32)
尝试输送100批次会引发错误:
ValueError: Cannot feed value of shape (100, 224, 224, 3) for Tensor 'input:0', which has shape '(1, 224, 224, 3)'
有没有办法绕过它?
P.S。
以下是我尝试训练的方法:
sess.run(train_step,
feed_dict={image_buffer_input: train_image_buffers,
ground_truth_input: train_ground_truth})