python - 如何在 tensorflow 会话中运行 Keras.model() 进行预测？

Question

我目前遇到问题，在 tensorflow 会话中执行我的模型预测 keras 时。

with tf.Session(graph=graph) as sess:
    sess.run(tf.global_variables_initializer())
    ## want to know how to add model.predict() inside this condition
    predictions = model.predict(#my_model)
    #predictions output is same not appending

或任何替代方法都会有所帮助。

任何帮助，将不胜感激。

标签： pythontensorflowkeras

Answer 1

from keras import backend as K

with tf.Graph().as_default():

    with tf.Session() as sess:

        K.set_session(sess)
        model = load_model(model_path)
        preds = model.predict(in_data)