将ckpt文件转换为冻结的推理图

时间:2019-08-07 07:53:28

标签: python tensorflow deep-learning computer-vision artificial-intelligence

我有检查点文件* .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

我要使用checkpoints.txt和* .data,*。index和* .meta创建冻结的推理图

我的主要挑战是从checkpoints.txt文件中找到输出节点名称。如果您看到此文件,则将显示名称列表,这些名称之间没有明确的分隔符,确切地说就是输出

一旦找到输出节点名称。创建冻结图很容易。 如何从checkpoints.txt文件中的名称列表中找到输出?

0 个答案:

没有答案