Keras中的ImageDataGenerator

Question

我正在使用Keras功能ImageDataGenerator()来生成培训和验证数据集。我试图了解这个功能在内部的作用。该功能执行的预处理步骤是什么？ 我在哪里可以找到这个函数的源代码？

Answer 1

您可以在以下网址找到任何源代码：

• https://github.com/keras-team/keras

这是ImageDataGenerator：

• https://github.com/keras-team/keras/blob/master/keras/preprocessing/image.py/#L374

keras文档页面还提供了引导您的链接：

• https://keras.io/preprocessing/image/

在内部，ImageDataGenerator将对您提供的图像进行一系列不同的数据扩充程序，并准备一个python generator供您在拟合模型时使用。

有几种数据增强方法可供使用，您可以在上面的帮助页面中了解它们的含义。

生成器用于在循环中创建批处理。在这种情况下，一次一批图像。

您可以model.fit()使用model.fit_generator()或ImageDataGenerator.flow()，而不是ImageDataGenerator.flow_from_directory()。