ImageDataGenerator形状问题

Question

我是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”的形状在哪里。因此，我在训练模型时遇到问题。

有什么建议吗？

Answer 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)。希望这会有所帮助