ImageDataGenerator in Keras



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

您可以在:

上找到任何源代码
  • https://github.com/keras-team/keras

这是Imagedatagenerator:

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

KERAS文档页面还具有引导您到达那里的链接:

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

在内部,ImageDataGenerator将在您提供的图像上制作一系列不同的数据增强过程,并准备python generator供您使用时使用。

使用几种数据增强方法,您可以了解它们在上面的帮助页面中的内容。

发电机用于在循环中创建批处理。在这种情况下,一次一次图像。

代替使用model.fit(),您将使用model.fit_generator()ImageDataGenerator.flow()ImageDataGenerator.flow_from_directory()

最新更新