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

落日映苍穹つ 2022-09-10 11:13 127阅读 0赞

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

> graph = tf.get\_default\_graph()

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA54ix6Zeu6KW_55Oc54ix5aSn5qCR_size_14_color_FFFFFF_t_70_g_se_x_16]  
2、在服务中调用模型进行识别前，加上这句：

> with graph.as\_default():

![在这里插入图片描述][7f19afba59c24f9982e0424330746028.png]

[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA54ix6Zeu6KW_55Oc54ix5aSn5qCR_size_20_color_FFFFFF_t_70_g_se_x_16]: /images/20220829/de6c0a0ce404457f9eb343518708adf5.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA54ix6Zeu6KW_55Oc54ix5aSn5qCR_size_14_color_FFFFFF_t_70_g_se_x_16]: /images/20220829/f5839c3aa7ce47d2996e463e82bd8f18.png
[7f19afba59c24f9982e0424330746028.png]: /images/20220829/dfa4160e67104f0a903d822cad6474f1.png