ImageDataGenerator形状问题

Question

我对凯拉斯很陌生。在使用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()中为flow_from_directory()提供一个值。

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

上述代码将生成形状(64,227,227,3)的批处理。希望这能有所帮助