python - 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它将对您提供的图像进行一系列不同的数据增强程序，并准备一个 pythongenerator供您在拟合模型时使用。

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

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

model.fit()您将使用model.fit_generator()withImageDataGenerator.flow()或代替 using ImageDataGenerator.flow_from_directory()。