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

Question

我正在使用 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呢？

Answer 1

您可以使用以下

data_gen.filenames