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)))

ValueError in input to CNN