Keras-在ImageDataGenerator中正确使用preprocess_input

时间:2018-12-28 15:47:58

标签: python keras

我正在尝试在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)

两种方法似乎都能产生明智的结果,但我不知道哪种方法更可取/正确?

0 个答案:

没有答案