代码之家  ›  专栏  ›  技术社区  ›  AaronDT

keras-在imagedatagenerator中正确使用预处理输入

  •  0
  • AaronDT  · 技术社区  · 7 年前

    我试图在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 回复  |  直到 7 年前