Tensor is not an element of this graph 解决方法

当两个Tensorflow模型在Flask服务中运行时出现错误,问题可能出在tensorflow图的管理上。解决方案是确保所有模型都在同一默认图下运行。在创建Flask服务后,获取默认图,并在调用模型前使用该图作为上下文。
摘要由CSDN通过智能技术生成

问题描述:
有两个模型,单独运行两个模型的pipeline时没有问题,但是封装到flask服务后报了如下错误:
在这里插入图片描述
解决方法:
以前一版遇到这种问题的时候都是模型加载错误,保存的模型和网络结构不符合导致。但是这次不是这个问题,因为单独运行pipiline是没有问题的。
后来经高手指点,发现是tensorflow图的问题。tensorflow是基于图运行的,在启动服务的时候,新建一个基础的默认图,让另外两个模型都基于这个默认图运行即可。
具体代码:
1、在创建flask服务后,加上一句:

graph = tf.get_default_graph()

在这里插入图片描述
2、在服务中调用模型进行识别前,加上这句:

with graph.as_default():

在这里插入图片描述

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值