如何为标签塑造图像[重复]

Question

我正在尝试训练卫星图像。我的训练集有 2 个输入文件夹（样本）和基本事实（标签）。

图片总数为：496
我的 X_train 形状是：(496,3,256,256)
y_train 形状为：(496,256,256)

对于每个模型，我都会收到关于密集层的错误。

ValueError: Expected dense to have 2 dimensions, but got an array with shape (496,256,256)

Answer 1

您正在将图像馈送到密集层。使用 .flatten() 展平图像或使用具有 CNN 层的模型。形状 (496,3,256,256) 表示您有 496 张图像，大小为 256x256，具有 3 个维度 (rgb)。

在这个answer 上查找 CNN 模型。

CNN: Error when checking input: expected dense to have 2 dimensions, but got array with shape (391, 605, 700, 3)