tensorflow模型文件(ckpt)转pb文件的方法（不知道输出节点名）

转换 TensorFlow 模型文件（ckpt）为 TensorFlow pb 文件的方法如下：

步骤1：确定输出节点名称

在转换过程中需要指定输出节点的名称。有两种方法可以确定 TF 模型中输出节点的名称。

方法1：查看已知的模型输出节点名称

如果你知道需要转化的节点名称，可直接跳到下一步骤。如果不知道，可以使用 TensorBoard 工具查看模型输出节点名称。启动 TensorBoard 并加载 modelo 便可以查看模型的节点名称：

tensorboard --logdir=path/to/model

然后在浏览器中打开 http://localhost:6006/，通过 Graphs 标签查看模型的节点信息。

方法2：使用 freeze_graph 工具

另一种确定输出节点名称的方法是使用 freeze_graph 工具。freeze_graph 工具会将 TF 模型文件中所有变量的数值恢复，并将模型图及其相应变量的数值存储到一个单独的文件中。在 freeze_graph 工具中，需要指定模型中需要输出的节点名称。

示例：

python freeze_graph.py \
  --input_graph=path/to/ckpt/model.pb \
  --input_checkpoint=path/to/ckpt/model.ckpt \
  --output_graph=frozen_model.pb \
  --output_node_names=output_node

在这个示例中，我们假设模型文件存储在 path/to/ckpt 目录下，ckpt 模型文件的名字为 model.ckpt，pb 模型文件的名字为 model.pb。由于不知道输出节点的名称，因此在使用 freeze_graph 工具前需要查看模型的节点信息。假设在模型文件中有一个输出节点 output_node，则可以使用 freeze_graph 工具将 ckpt 模型文件转化成 pb 模型文件，并指定输出节点名称为 output_node。

步骤2：使用 convert_variables_to_constants 将变量转化成常量

在载入模型之后，需要将模型中的变量转化成常量，以便能够轻松地在其他设备上运行模型。

示例：

import tensorflow as tf

# 加载模型
saver = tf.train.import_meta_graph('path/to/model.ckpt.meta')
graph = tf.get_default_graph()

# 将变量转成常量
output_graph_def = tf.graph_util.convert_variables_to_constants(
      sess,
      graph.as_graph_def(),
      output_node_names.split(','))

在这个示例中，我们首先通过 tf.train.import_meta_graph() 函数加载模型。然后，我们将变量转化成常量，并将常量保存到 output_graph_def 变量中。

步骤3：将常量图写入 pb 文件

最后，我们将常量图写入文件中以获得最终的模型文件。

示例：

# 写入 pb 文件
with tf.gfile.GFile(output_graph, "wb") as f:
    f.write(output_graph_def.SerializeToString())

在这个示例中，我们使用 tf.gfile.GFile() 函数将常量图写入文件，该函数在 TensorFlow 中被用于文件操作。

现在，我们已经演示了将 TensorFlow ckpt 模型文件转化成 pb 模型文件的完整过程。以下是另外一个示例：

import tensorflow as tf

# 加载模型
saver = tf.train.import_meta_graph('path/to/model.ckpt.meta')
graph = tf.get_default_graph()

# 将变量转成常量
output_graph_def = tf.graph_util.convert_variables_to_constants(
    sess,
    graph.as_graph_def(),
    ['output_node'])

# 写入 pb 文件
with tf.gfile.GFile('path/to/frozen_model.pb', "wb") as f:
    f.write(output_graph_def.SerializeToString())

在这个示例中，我们将输出节点的名称设置为 output_node，将常量图保存到 name 为 frozen_model.pb 的文件中。