Keras 中的图像数据生成器

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() 和 ImageDataGenerator.flow() 或 ImageDataGenerator.flow_from_directory()。