Keras: Как использовать метод flow_from_directory класса ImageDataGenerator для модели с несколькими выходами - PullRequest
0 голосов
/ 24 января 2020

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

Как создать собственный генератор, который будет возвращать объект генератора в формате [input_x], [logits_variance,softmax] с использованием метода flow_from_directory?

Я включил скриншот структуры модели, чтобы лучше понять, что я имею в виду.

enter image description here

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...