首先把错误贴出来~
ValueError: Tensor Tensor("dense_1/Sigmoid:0", shape=(?, ?, 1), dtype=float32) is not an element of this graph
错误场景:
使用flask部署tensorflow 模型进行预测的api时,莫名报错如上,首先排除了部署代码问题,因为在主函数里是预测成功的,但是在flask的api函数中就会报错
解决办法:
在你构造或load完的模型后(紧接着),保存 TensorFlow graph:
graph = tf.get_default_graph()
当你使用model进行预测的时候外面套一层:
global graph
with graph.as_default():
(... do inference here ...)