keras fit_generator多输入，多输出

最新推荐文章于 2022-10-09 10:55:30 发布

草地小猪

最新推荐文章于 2022-10-09 10:55:30 发布

阅读量783

点赞数

文章标签： keras 深度学习人工智能

本文链接：https://blog.csdn.net/weixin_42348282/article/details/125545340

版权

history = model.fit_generator(
        generator=training_data,
        steps_per_epoch=int(len(training_data) * train_step_multiplier),
        validation_data=validation_data,
        validation_steps=int(len(validation_data) * val_step_multiplier),
        epochs=epochs,
        verbose=1,  # 日志显示模式 0 = 安静模式, 1 = 进度条, 2 = 每轮一行。
        callbacks=callbacks

training_data = ({'input_1': crops, 'input_2': x}, {'output': labels})#和模型中的name对应

input_1 = keras.layers.Input((n_crops, n_crops) + crop_shape,name='input_1')#7*7*16*16*3
input_2 = keras.layers.Input(img_shape,name='input_2')
x = keras.layers.Dense(units=n_classes, activation='sigmoid',name='output')(x)
model = keras.models.Model(inputs=[input_1, input_2], outputs=x)