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()结合使用。