2024-07-08 07:53:35 发布
网友
我该如何从像model.ckpt.meta这样的metagraphdef文件生成graph.pbtxt,即包含graph def的protobuf?在
model.ckpt.meta
graph.pbtxt
我想这两个文件,graph.pbtxt和{}都是在培训期间生成的。但是,我想使用一个分析工具,它需要一个graph.pbtxt在一个预先训练好的NN上,我只能访问model.ckpt.*文件。这可能吗?在
model.ckpt.*
要生成protobuf,需要两件事:
.ckpt
本质上,.meta文件包含图形定义,而.ckpt包含训练模型的权重。protobuf格式将二者合并为一个文件。在
.meta
您必须加载元图,然后加载检查点,如下所示:
sess = tf.Session() saver = tf.train.import_meta_graph(meta_file) saver.restore(save_path=ckpt_file, sess=sess)
最后,在将图形导出为protobuf格式之前,必须定义图形的输出节点。输出节点包含在要检索的输出层的名称列表中。例如,它可以是['myModel/fc12/BiasAdd']。在
['myModel/fc12/BiasAdd']
要生成protobuf,需要两件事:
.ckpt
。在本质上,
.meta
文件包含图形定义,而.ckpt
包含训练模型的权重。protobuf格式将二者合并为一个文件。在您必须加载元图,然后加载检查点,如下所示:
最后,在将图形导出为protobuf格式之前,必须定义图形的输出节点。输出节点包含在要检索的输出层的名称列表中。例如,它可以是
^{pr2}$['myModel/fc12/BiasAdd']
。在相关问题 更多 >
编程相关推荐