带有 Tensorflow 后端的 Keras 中的图形会话空错误

Question

我正在使用如下所示的代码子集测试我更大的图像识别模型。理想情况下，它应该能够加载 imagenet 模型，然后显示“hello world”，但它会在 graph = K.get_session().graph 行抛出错误，即 RuntimeError: The Session graph is empty。在调用 run() 之前向图形添加操作。 我尝试更改线条的位置，但它不起作用。我想在我的主代码中使用with.graph as default()。我正在使用 keras 2.09。

import dlib
import requests
import numpy as np
from skimage import io
from skimage.transform import resize
from keras import backend as K
from keras.models import Sequential
from keras.layers import Dropout, Flatten, Dense
from keras import applications
from flask import Flask, jsonify, request, abort, make_response
app = Flask(__name__)
auth_token = 'WyIxYSDFg467YT.A3MmJlODcyODkzOGQzZjk4YzUiXQ.B5e5SgsDcaMgiRqx21Ydf8M'
top_model_weights_ethnicity = 'ethnicity.071217.23-0.28.hdf5'
img_width, img_height = 139, 139
confidence_ethnicity = '0.59'
detector = dlib.get_frontal_face_detector()
graph = K.get_session().graph
def get_pretrained_model():
    with graph.as_default():
        pretrained_model = applications.InceptionResNetV2(include_top=False, weights='imagenet',
                                                          input_shape=(img_width, img_height, 3))
        return pretrained_model
with graph.as_default():
    pretrained_model = get_pretrained_model()
@app.route("/")
def hello():
    return "<h1 style='color:blue'>Hello There!</h1>"
if __name__ == "__main__":
    app.run(host='0.0.0.0')

Answer 1

错误是由于我用于我的程序的已弃用的 Keras 版本造成的。我将它升级到最新版本，即 2.2.4，它为我解决了这个问题。