0

I am using keras.preprocess.image.ImageDataGenerator.flow_from_directory to generate image batches. However, when I try to use model.predict_generator to predict the labels, I do not know the label and the correspond image_id. Anyone help me? My code is as follows:

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))

I can get the result label, but how to get the correspond iamge_id?

Ioannis Nasios
  • 8,292
  • 4
  • 33
  • 55
Abraham Ben
  • 147
  • 1
  • 2
  • 12

1 Answers1

2

You can use the following 

data_gen.filenames
Ioannis Nasios
  • 8,292
  • 4
  • 33
  • 55