最近在做圖像分類的任務,使用了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) # 按順序輸出文件的名字