如何为标签塑造图像

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)