小编jon*_*ava的帖子

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

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

python conv-neural-network keras

2
推荐指数
1
解决办法
2808
查看次数

标签 统计

conv-neural-network ×1

keras ×1

python ×1