python - 当图像尺寸为(2048 * 1536)时，我应该在顺序模型的第一层采用什么输入形状

Question

我有一个图像数据集，每个图像的尺寸为 (2048,1536)。在 ImageDataGenerator 中从目录中获取数据，我使用了相同的目标大小，即 (2048,1536)，但是在制作顺序模型第一层时，我应该使用什么输入形状？它是否与 (2048,1536) 相同，或者我可以采取任何随机形状，例如 (224,224)。

Answer 1

您可能应该通过制作一个大小为 3145728 (2048 * 1536) 的向量来展平您的输入数据。如果您的数据在 numpy 数组中，您可以使用 np.flatten() ( numpy flatten )。

然后您的第一层可以具有与此矢量相同的形状。