我正在使用我自己的圖像數據生成器。它生成圖像批次的 0、90、180 和 270 度旋轉版本，并返回它們的類。我使用內置ImageDataGenerator函數來測試模型。但是會flow_from_directory生成不同的類索引。的輸出train_generator.class_indices是{'0': 0, '90': 1, '180': 2, '270': 3}。但test_generator.class_indices返回{'0': 0, '180': 1, '270': 2, '90': 3}。只是我可以改變旋轉角度的順序，但這個問題是由操作系統的文件系統引起的，我將在不同的操作系統上運行代碼。在這種情況下，我需要一個自動化的解決方案。有沒有辦法改變flow_from_directory方法的類索引？
2 回答
慕后森
TA貢獻1802條經驗 獲得超5個贊
看起來你可以做到
flow_from_directory(directory,
classes={'0': 0,
'90': 1,
'180': 2,
'270': 3}
)
添加回答
舉報
0/150
提交
取消