Как использовать keras ImageDataGenerator flow_from_directory, учитывая карту от имени изображения до метки класса?

Question

Структура папки выглядит следующим образом:

-dataset
   |---train_set
   |      |-- img01.jpg
   |      |-- img02.jpg
   |      |-.....
   |
   |---val_set 

У меня нет вложенных папок в train_set, имя которых соответствует метке класса. Однако у меня есть словарь от имени изображения до ярлыка класса, например map = {'img01.jpg': 10, 'img02.jpg': 3, 'img03.jpg':12, ...}

Как использовать keras 'ImageDataGenerator.flow_from_directory в этом случае, так как он требует, чтобы я сортировал изображения в разных подпапках? Объем моего тренировочного набора составляет около 500 G, не удобно создавать подпапки в соответствии с ярлыками классов. Если я не могу использовать ImageDataGenerator.flow_from_directory, есть ли альтернативный способ реализовать эту функцию?

Answer 1

Класс Keras ImageDataGenerator также предлагает метод flow_from_dataframe () , который должен выполнять эту работу.

Посмотрите этот учебник (вторая часть) для вашего примера кейса.