Как получить соответствующий image_id и метку при использовании предиката_генератора и потока_данных

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