ImageDataGenerator 形状问题

【问题标题】：ImageDataGenerator Shape IssueImageDataGenerator 形状问题
【发布时间】：2018-08-20 00:36:15
【问题描述】：

我是 Keras 的新手。我在使用 ImageDataGenerator 时遇到了一些形状问题。以下是我的代码：

train_batches = ImageDataGenerator().flow_from_directory(trainpath,target_size(227,227),classes=['class1','class2'])

x_train, y_train = next(train_batches)

当我打印 x_train 的形状时，它给出了(32, 227, 227, 3)。我不确定“32”在哪里出现。因此，我在训练模型时遇到了问题。

有什么建议吗？

【问题讨论】：

32 是批量大小。

标签： python-3.x keras

【解决方案1】：

方法flow_from目录中有一个参数叫做batch_size，你可以在这里了解更多
https://keras.io/preprocessing/image/

所以如果你想改变生成的训练批次的形状，你可以在 flow_from_directory() 中给 batch_size 一个值

train_batches = ImageDataGenerator().flow_from_directory(trainpath,target_size(227,227),batch_size=64,classes=['class1','class2'])

以上代码会生成批量形状(64,227,227,3)。希望这会有所帮助

【讨论】：