代码之家  ›  专栏  ›  技术社区  ›  Abraham Ben

使用predict\u生成器和flow\u时,如何从目录中获取相应的image\u id和标签

  •  0
  • Abraham Ben  · 技术社区  · 7 年前

    我正在使用keras。预处理。形象图像数据生成器。从_目录流_以生成图像批。然而,当我尝试使用模型时。predict\u generator要预测标签,我不知道标签和对应的image\u id。有人帮我吗?我的代码如下:

    from keras.preprocess.image import ImageDataGenerator
    
    from keras.applications.vgg19 import VGG19
    base_model = VGG19(...)
    data_gen = ImageDataGenerator(...)
    
    result = model.predict_generator(date_gen.flow_from_directory(valid_path), shuffle=False, target_size=(input_size, input_size))
    

    我可以获得结果标签,但如何获得相应的iamge\u id?

    1 回复  |  直到 7 年前
        1
  •  2
  •   Ioannis Nasios    7 年前

    您可以使用以下选项

    data_gen.filenames