是否可以从目录更改 Keras 流的类索引

Question

我正在使用自己的图像数据生成器。它生成图像批次的 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方法的类索引？

Answer 1

看起来你可以做

flow_from_directory(directory, 
                    classes={'0': 0, 
                             '90': 1, 
                             '180': 2, 
                             '270': 3}
                   )