Keras: How to use fit_generator with multiple outputs of different type

后端 未结 3 608
春和景丽
春和景丽 2020-12-05 05:21

In a Keras model with the Functional API I need to call fit_generator to train on augmented images data using an ImageDataGenerator. The problem is my model has two outputs:

3条回答
  •  陌清茗
    陌清茗 (楼主)
    2020-12-05 06:03

    The best way to achieve this seems to be to create a new generator class expanding the one provided by Keras that parses the data augmenting only the images and yielding all the outputs.

提交回复
热议问题