当从protobuf加载时,张量名称是否总是以“import /”为前缀?

时间:2017-06-13 10:22:57

标签: tensorflow

当我使用以下方法从磁盘加载(冻结)Tensorflow模型时

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)

似乎所有张量名称都以import /为前缀。  这是我用来打印名称的代码:

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 个答案:

答案 0 :(得分:2)

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