2017-06-13 21 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以使用不同的前綴。