TF会话图为空-从两个导入的图创建新的tf图

时间:2018-11-13 18:42:31

标签: python tensorflow machine-learning computer-vision

现在,我正在使用两个模型,其中一个是从检查点加载的,另一个是加载冻结的模型的。在一个类中,我有一个loadmodel函数,另一个处理函数是要将这两个模型都应用到图像上,以便将来的程序只需调用该类的方法就可以轻松地处理图像上的所有内容。 / p>

这是代码的简化示例,仅用于我在类中苦苦挣扎的方法:

raise RuntimeError('The Session graph is empty.  Add operations to the '
RuntimeError: The Session graph is empty.  Add operations to the graph before calling run().

我上面得到的错误是

self.g1 = self.seg_graph.as_graph_def()
self.g2 = self.eval_graph.as_graph_def()
seg_output, = tf.import_graph_def(self.g1, input_map = {"ImageTensor:0" : resized_img, "img_placeholder:0" : resized_img}, return_elements = ["SemanticPredictions:0"])
style_output, = tf.import_graph_def(self.g2, input_map={"img_placeholder:0" : resized_img}, return_elements=["style_outputs:0"])

我也尝试过在with语句中包装合并的会话,同时尝试嵌套和外部,但尚未使它起作用。

我尝试的另一件事是代替get_tensor_by_name:

ValueError: Input 0 of node import/Shape was passed double from import/_inputs/Const:0 incompatible with expected uint8.

但这会返回错误:

{{1}}

任何有识之士都将不胜感激,谢谢!还是tf :)的新手。

0 个答案:

没有答案