Можно ли изменить индексы классов потока 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}
                   )