在Django中加载多个keras重量文件

时间:2019-03-25 12:24:40

标签: django tensorflow keras

我有一个Django Web应用程序,需要与两个Keras模型集成。我有权重文件(.hdf5)文件,我建立了基线模型,然后加载权重,如下所示。每当我尝试使用feed a value to Tensor进行预测时,我都会得到它。我已将错误与屏幕截图一同添加

我尝试了具有不同图形和会话逻辑的各种解决方案,但没有一个起作用。我正在尝试的当前版本,出现错误


def build_model():
    model = tf.keras.models.Sequential()

    model.add(tf.keras.layers.Flatten(input_shape=(28,28,)))
    model.add(tf.keras.layers.Dense(32, activation="relu"))
    model.add(tf.keras.layers.Dense(10, activation="sigmoid"))

    return model

def build_model_1():
    model = tf.keras.models.Sequential()

    model.add(tf.keras.layers.Flatten(input_shape=(256,256,)))
    model.add(tf.keras.layers.Dense(32, activation="relu"))
    model.add(tf.keras.layers.Dense(32, activation="relu"))
    model.add(tf.keras.layers.Dense(10, activation="sigmoid"))

    return model

graph1 = Graph()
    with graph1.as_default():
        model = build_model()
        model.load_weights('model_weights.hdf5')
        model.compile(optimizer.....)
        model._make_predict_function()
graph2 = Graph()
    with graph2.as_default():
        model_1 = build_model_1()
        model_1.load_weights('model_weights_1.hdf5')
        model_1.compile(optimizer.....)
        model_1._make_predict_function()

Error image

0 个答案:

没有答案