输入到CNN的ValueError

Question

使用python中的keras制作图像分割模型，但在第一层获取Conv2D时出错。 在代码中，这一行如下：

model.add(Conv2D(32,kernel_size=(3, 3),padding='same',input_shape=(1,500,366,3)))

但是收到以下错误：

ValueError: Input 0 is incompatible with layer conv2d_1: expected ndim=4, found ndim=5

我最初是为一张图片制作这个模型。

Answer 1

尝试删除您的第一个维度：

model.add(Conv2D(32,kernel_size=(3, 3),padding='same',input_shape=(500,366,3)))