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

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

graph = tf.get_default_graph()

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

with graph.as_default():

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值