InvalidArgumentError：输入必须是4维[8,6171,4]

Question

我正在运行一个2d卷积网络。我的输入有3个维度，但我得到这个4维错误：dimension error

如您所见，我的输入具有正确的尺寸：

correct input dimension

这是我的代码：

from keras import models
from keras import layers
model = models.Sequential()
model.add(layers.Conv2D(32, (3, 3), activation='relu', input_shape=(8,6171,4)))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(64, (3, 3), activation='relu'))

model.add(layers.Flatten())
model.add(layers.Dense(64, activation='relu'))
model.add(layers.Dense(10, activation='softmax'))

当我只构建一个3-D输入层时，为什么它要求我进行4-D？

请帮忙。谢谢。

Answer 1

你需要使用

new_image = tf.expand_dims(image,0)

因为模型需要数据集而不是单个图像。