是否可以从目录中更改 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}

                   )