我有一个图像数据集,每个图像的尺寸为((2048,1536))。在ImageDataGenerator以从目录中获取数据时,我使用了相同的目标大小,即(2048,1536),但是在使顺序模型成为第一层时,输入了什么形状我应该使用吗?它与(2048,1536)相同还是可以采用(224,224)之类的任意随机形状?

最佳答案

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

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

10-08 05:01