keras從數據生成器ImageDataGenerator中的flow_from_directory中獲取加載的文件名和相應的類別

最近在做圖像分類的任務，使用了densenet進行fine-tune，因爲圖片數據比較多且佔內存因此數據的加載使用ImageDataGenerator生成器，使用flow_from_directory從文件夾中獲取各個類別的數據。

因爲在測試的時候，需要知道哪個數據被判錯了，要找到對應的文件名。

test_datagen = ImageDataGenerator(rescale=1./255)
val_generator = test_datagen.flow_from_directory( test_dir,
                                                  target_size=(img_size, img_size),
                                                  batch_size=32,
                                                  shuffle=False)
print(val_generator.class_indices)   # 輸出對應的標籤文件夾
print(val_generator.filenames)  # 按順序輸出文件的名字