我正在使用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()
。