ValueError: Tensor Tensor xxx is not an element of of this graph的解决方案

问题：

在利用keras+flask构建一个简单的深度学习后台服务器，遇到了如下的问题：

ValueError: Tensor Tensor("fc1000/Softmax:0", shape=(?, 1000), dtype=float32) is not an element of this graph。

 

解决方案如下：

在初始化的时候，加载模型文件和生成graph。

 

完整代码如下：

# USAGE
# Start the server:
# 	python run_keras_server.py
# Submit a request via cURL:
# 	curl -X POST -F image=@dog.jpg 'http://localhost:5000/predict'
# Submita a request via Python:
#	python simple_request.py

# import the necessary packages
from keras.applications import ResNet50
from keras.preprocessing.image import img_to_array
from keras.applications import imagenet_utils
from PIL import Image
import numpy as np
import flask
import io
import tensorflow as tf

# initialize our Flask application and the Keras model
app = flask.Flask(__name__)

graph = None
model = None


def load_model():
    # load the pre-trained Keras model (he