Функциональная модель с несколькими ветвями

Question

Я пытаюсь добавить несколько веток классификации поверх сети InceptionV3. Я использую imageGenerator, но не могу настроить его правильно.

У меня есть один imageGenerator для обеих веток, который читает данные из одного корневого каталога.

Мои корневые каталоги (train, test) имеют следующую структуру:

• класс а (например, ткань)

  • Хлопок
  • Кожа

• класс b (например, цвет)

  • черный
  • синий

Код, который я использую для инициализации генераторов:

train_generator = train_datagen.flow_from_directory('./train',
                                                   target_size = target_size,
                                                   batch_size = batch_size,
                                                   class_mode = 'categorical',
                                                   shuffle=True)

test_generator = test_datagen.flow_from_directory('./test',
                                                   target_size = target_size,
                                                   batch_size = batch_size,
                                                   class_mode = 'categorical',
                                                   shuffle=True)

Я не уверен, как бы я обучал каждую ветку в моей сети, используя данные обучения для этого класса.

Answer 1

Вы хотите обучить сеть для классификации по нескольким меткам.

Вам не нужно , чтобы иметь несколько веток; Вы можете обучить только одну нейронную сеть, чтобы узнать цвет и тип одежды.

Мое предложение - взглянуть на этот блог, это именно то, что вам нужно:

https://www.pyimagesearch.com/2018/05/07/multi-label-classification-with-keras/

Обратите внимание, что вам может понадобиться реструктурировать тренировочный набор, например, иметь папку с синим платьем, другую папку с синими джинсами, затем папку с красным платьем и т. Д.

Однако, если вам нужно иметь несколько веток, вы обязательно должны взглянуть на это: https://www.pyimagesearch.com/2018/06/04/keras-multiple-outputs-and-multiple-losses/