如何在使用 predict_generator 和 flow_from_directory 时获取对应的 image_id 和标签
how to get the correspond image_id and label when using predict_generator and flow_from_directory
我正在使用 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
我正在使用 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