0

keras.preprocess.image.ImageDataGenerator.flow_from_directoryを使用して画像バッチを生成しています。しかし、ラベルを予測するためにmodel.predict_generatorを使用しようとすると、ラベルと対応するimage_idがわかりません。誰でも助けてくれますか?次のように私のコードは次のとおりです。predict_generatorとflow_from_directoryを使用する場合の対応する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を取得する方法?

答えて

2

data_gen.filenames 
を使用することができます
関連する問題