我有一个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()