4

我正在使用烧瓶运行 Web 服务器，当我尝试使用 vgg16 时出现错误，这是 keras 预训练的 VGG16 模型的全局变量。我不知道为什么这个错误会上升，或者它是否与 Tensorflow 后端有关。这是我的代码：

vgg16 = VGG16(weights='imagenet', include_top=True)

def getVGG16Prediction(img_path):
    global vgg16

    img = image.load_img(img_path, target_size=(224, 224))
    x = image.img_to_array(img)
    x = np.expand_dims(x, axis=0)
    x = preprocess_input(x)

    pred = vgg16.predict(x)
    return x, sort(decode_predictions(pred, top=3)[0])

@app.route("/uploadMultipleImages", methods=["POST"])
def uploadMultipleImages():
    uploaded_files = request.files.getlist("file[]")
    for file in uploaded_files:
        path = os.path.join(STATIC_PATH, file.filename)
        pInput, result = getVGG16Prediction(path)

这是完整的错误： 在此处输入图像描述

非常感谢任何意见或建议。谢谢你。

4

1 回答 1

2

看看avital关于这个 github 问题的答案。在这里引用相关部分：

加载或构建模型后，立即保存 TensorFlow 图：

graph = tf.get_default_graph()

在另一个线程中（或者可能在异步事件处理程序中），执行以下操作：

global graph
with graph.as_default():
    (... do inference here ...)

我对此进行了一些修改并将图形存储在我的应用程序的配置对象中，而不是使其成为全局对象。

TensorFlow 文档解释get_default_graph了为什么这是必要的：

注意：默认图是当前线程的属性。如果您创建一个新线程，并希望在该线程中使用默认图形，则必须在该线程的函数中显式添加一个 with g.as_default():。

于 2018-06-18T21:40:32.450 回答