python - Keras - 在 ImageDataGenerator 中正确使用 preprocess_input

Question

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

但是，我不确定如何keras.applications.inception_v3.preprocess_input在ImageDataGenerator

此外，我发现了两种方法：

1)

datagen=ImageDataGenerator(preprocessing_function=keras.applications.inception_v3.preprocess_input)

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)

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

标签： pythonkeras