0

我正在尝试使用 TensorFlow 从 VGG 网络中提取特征。我正在以形状(1000、448、448、3)的 numpy 数组加载一批 1000 张图像,并尝试使用 feed_dict 为网络提供这批图像:

pool = sess.run(vgg.lastpool, feed_dict={vgg.imgs: [image_list]})

其中 image_list 是上述形状数组 (1000, 448, 448, 3)。

但是,我收到以下错误:

ValueError: Cannot feed value of shape (1, 1000, 448, 448, 3) for Tensor u'Placeholder:0', which has shape '(?, 448, 448, 3)'

当我尝试使用单个图像(形状 = (1, 448, 448, 3))时,它可以工作,但是当我尝试提供一批时,它会添加这个随机维度。我尝试使用 numpy 的挤压,但同样的事情发生了......在使用 feed_dict 之前,我检查了 image_list 的形状肯定是(1000、448、448、3)。

4

0 回答 0