machine-learning - 在使用predict_generator和flow_from_directory时如何获取对应的image_id和标签

我正在使用keras.preprocess.image.ImageDataGenerator.flow_from_directory生成图像批处理。但是，当我尝试使用model.predict_generator来预测标签时，我不知道标签和对应的image_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_id？

最佳答案

您可以使用以下内容

data_gen.filenames

关于machine-learning - 在使用predict_generator和flow_from_directory时如何获取对应的image_id和标签，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/47071263/