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