从中提取图形定义型号.ckpt.米

2024-07-08 07:53:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我该如何从像model.ckpt.meta这样的metagraphdef文件生成graph.pbtxt,即包含graph def的protobuf?在

我想这两个文件,graph.pbtxt和{}都是在培训期间生成的。但是,我想使用一个分析工具,它需要一个graph.pbtxt在一个预先训练好的NN上,我只能访问model.ckpt.*文件。这可能吗?在


Tags: 文件工具modeldefnnmetagraphprotobuf
1条回答
网友
1楼 · 发布于 2024-07-08 07:53:35

要生成protobuf,需要两件事:

  • {cd1>你提到
  • 旁边生成的检查点.ckpt。在

本质上,.meta文件包含图形定义,而.ckpt包含训练模型的权重。protobuf格式将二者合并为一个文件。在

您必须加载元图,然后加载检查点,如下所示:

sess = tf.Session()
saver = tf.train.import_meta_graph(meta_file)
saver.restore(save_path=ckpt_file, sess=sess)

最后,在将图形导出为protobuf格式之前,必须定义图形的输出节点。输出节点包含在要检索的输出层的名称列表中。例如,它可以是['myModel/fc12/BiasAdd']。在

^{pr2}$

相关问题 更多 >

    热门问题