2017-06-13 35 views
0

当我从盘使用加载(冻)Tensorflow型号:当从protobuf加载时,张量名称是否始终预先加上“import /”?

graph = tf.Graph() 
    with graph.as_default(): 
     f = gfile.FastGFile("frozen_graph.pb", "rb") 
     graph_def = tf.GraphDef() 
     graph_def.ParseFromString(f.read()) 
     tf.import_graph_def(graph_def) 

这一切似乎张量名称与进口/前缀。 这是我使用的代码打印的名字:

with tf.Session(graph=graph) as sess: 
     all_ops = sess.graph.get_operations() 
     op_values = [op.values() for op in all_ops] 
     for values in op_values: 
      for each in value: 
       print each.name 

为什么?这是一种可以重写的默认选项吗?或者我可以在我的代码中依靠这个吗?我找不到任何文件,任何人都可以指出我对此有何参考?

回答

1

这是在tf.import_graph_def中指定的默认值。
您可以传递参数name以使用不同的前缀。