当我使用以下方法从磁盘加载(冻结)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
为什么呢?这是某种可以覆盖的默认选项吗?或者我可以在我的代码中依赖于此吗?我无法找到这个记录的任何人,有人可以指点我参考吗?