Я создал модель с несколькими выходами Keras (на самом деле она имеет два выхода). Также у меня есть хорошо отобранный набор данных изображений (каталог обучающих данных, содержащий один подкаталог на класс изображений, заполненный изображениями .png или .jpg), который я хочу передать модели во время обучения, используя метод flow_from_directory
из ImageDataGenerator
учебный класс. Однако по умолчанию этот класс возвращает DirectoryIterator
, что дает кортежи (x, y)
.
Как создать собственный генератор, который будет возвращать объект генератора в формате [input_x], [logits_variance,softmax]
с использованием метода flow_from_directory
?
Я включил скриншот структуры модели, чтобы лучше понять, что я имею в виду.
