我正在尝试在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)
两种方法似乎都能产生明智的结果,但我不知道哪种方法更可取/正确?