依次运行多个 Keras 模型答案

【问题标题】：Run multiple Keras models in sequence依次运行多个 Keras 模型
【发布时间】：2018-12-09 02:28:20
【问题描述】：

我在一个文件夹中的所有图像上依次运行 3 个 keras cnn 模型。在一张图像上完成所有三个模型的预测后，我在循环的下一次迭代中低于错误。

  File "/home/ubuntu/.local/lib/python2.7/site-packages/tensorflow/python/client/session.py", line 1078, in _run
    'Cannot interpret feed_dict key as Tensor: ' + e.args[0])
TypeError: Cannot interpret feed_dict key as Tensor: Tensor Tensor("input_1:0", shape=(?, ?, ?, 3), dtype=float32) is not an element of this graph.

我的一个模型的代码结构：

def model_1():

    K.clear_session()

    cnn_model = load_model(model_path, compile=False)


    with K.get_session().as_default() as sess:
        ...Do inference....

我已经尝试过here 和here 提到的解决方案，但没有一个对我有用。

【问题讨论】：

标签： python tensorflow keras

【解决方案1】：

添加 tf.Session().as_default() 解决了这个问题：

def model_1():

    K.clear_session()

    cnn_model = load_model(model_path, compile=False)

    tf.Session().as_default()

    with K.get_session().as_default() as sess:
        ...Do inference....

添加对我有用的答案，以防有人想解决类似问题。

【讨论】：