我有检查点文件* .data,*。index和* .meta。我必须将这些检查点文件转换为冻结的推理图。我见过this reference。您可以从here
下载我的检查点文件这里是我做的:
步骤1: 使用以下代码显示图形的内部层:
import tensorflow as tf
saver = tf.train.import_meta_graph('./VGGnet_fast_rcnn_iter_50000.meta', clear_devices=True)
graph = tf.get_default_graph()
input_graph_def = graph.as_graph_def()
sess = tf.Session()
saver.restore(sess, "./VGGnet_fast_rcnn_iter_50000")
for n in tf.get_default_graph().as_graph_def().node:
print(n.name)
writer = tf.summary.FileWriter('./log/', sess.graph)
这是输出:
我要使用checkpoints.txt和* .data,*。index和* .meta创建冻结的推理图
我的主要挑战是从checkpoints.txt文件中找到输出节点名称。如果您看到此文件,则将显示名称列表,这些名称之间没有明确的分隔符,确切地说就是输出
一旦找到输出节点名称。创建冻结图很容易。 如何从checkpoints.txt文件中的名称列表中找到输出?