首页 > 解决方案 > ImageDataGenerator.fit() 函数在 Keras 图像预处理中的作用是什么?

问题描述

有 ImageDataGenerator.flow() 函数,我们在其中传递训练图像并返回增强图像。但是 ImageDataGenerator.fit() 函数有什么作用呢?

标签: kerasimage-preprocessingdata-augmentation

解决方案


对于一些转换,例如居中或缩放,ImageDataGenerator 需要对数据进行统计,例如特征均值、标准差等。该fit()方法收集这些统计信息。这个 API 类似于scikit-learn中的预处理函数的工作方式。


推荐阅读