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

最新推荐文章于 2023-06-06 20:20:07 发布

原创

最新推荐文章于 2023-06-06 20:20:07 发布 · 置顶 · 6.6k 阅读

5 ·

CC 4.0 BY-SA版权

在尝试使用keras与flask构建深度学习服务器时遇到ValueError，提示Tensor不在图中。解决方案是确保在初始化时加载模型文件并创建graph。可以参考相关GitHub项目和博客文章来实施修复。

问题：

在利用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