Keras - 在 ImageDataGenerator 中正确使用 preprocess_input

Aar*_*nDT 10 python keras

我正在尝试微调 keras 中的 Inception-V3 模型。因此,我想使用内置的预处理函数和flow_from_dataframe.

但是,我不确定如何keras.applications.inception_v3.preprocess_inputImageDataGenerator

此外,我找到了两种方法:

1)

datagen=ImageDataGenerator(preprocessing_function=keras.applications.inception_v3.preprocess_input)
Run Code Online (Sandbox Code Playgroud)

2)

def preprocess_input_new(x):
   img = keras.applications.inception_v3.preprocess_input(img_to_array(x))
   return image.array_to_img(img)

datagen=ImageDataGenerator(preprocessing_function=preprocess_input_new)
Run Code Online (Sandbox Code Playgroud)

两种方式似乎都会产生合理但不同的结果,因此我想知道哪一种是首选/正确的?