【发布时间】:2021-08-11 07:53:27
【问题描述】:
我已经为相同的类标签训练了两个具有不同数据集的 keras 模型。我怎样才能将模型 keras_model.h5 和 keras_model2.h5 集成在一起并制作另一个 keras 模型说 keras_ensemble.h5。我试过参考各种互联网资源,但不是运气。 有人可以帮我提供集成它的代码吗? Here are the models I've trained
请帮助我完成这个。谢谢。
编辑: 这是我的代码,在回答我的问题 Frightera 的人的帮助下我得以完成
import tensorflow.keras
import tensorflow as tf
from PIL import Image, ImageOps
import numpy as np
# Disable scientific notation for clarity
np.set_printoptions(suppress=True)
# Load the model
keras_model = tensorflow.keras.models.load_model('keras_model.h5', compile=False)
keras_model._name = 'model1'
keras_model2 = tensorflow.keras.models.load_model('keras_model2.h5', compile=False)
keras_model2._name = 'model2'
models = [keras_model, keras_model2]
#model_input = tf.keras.Input(shape=(125, 125, 3))
model_input = tf.keras.Input(shape=(224, 224, 3))
model_outputs = [model(model_input) for model in models]
ensemble_output = tf.keras.layers.Average()(model_outputs)
ensemble_model = tf.keras.Model(inputs=model_input, outputs=ensemble_output)
编辑 如何获得 h5 格式的 keras 集成模型?
【问题讨论】:
标签: python machine-learning keras ensemble-learning image-classification