如何在 Keras 模型中获取 tensorflow 输出节点的名称？

Question

我正在尝试从我的 Keras（tensorflow 后端）模型创建一个 pb 文件，以便我可以在 iOS 上构建它。我正在使用 freeze.py，我需要传递输出节点。如何获取我的 Keras 模型的输出节点的名称？

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py

原文由 agsolid 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

您可以使用 Keras model.summary() 获取最后一层的名称。

如果 model.outputs 不为空，您可以通过以下方式获取节点名称：

 [node.op.name for node in model.outputs]

你通过

session = keras.backend.get_session()

然后通过以下方式将所有训练变量转换为常量

min_graph = convert_variables_to_constants(session, session.graph_def, [node.op.name for node in model.outputs])

之后你可以通过写一个protobuf文件

tensorflow.train.write_graph(min_graph, "/logdir/", "file.pb", as_text=True)

原文由 BerndSchmitt 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

如果在 Keras 中构建模型时没有明确指定输出节点，可以这样打印出来：

[print(n.name) for n in tf.get_default_graph().as_graph_def().node]

然后你需要做的就是找到合适的，这通常类似于激活函数的名称。您可以将找到的这个字符串名称用作 output_node_names 在 freeze_graph 函数中的值。

原文由 Oleg F. 发布，翻译遵循 CC BY-SA 3.0 许可协议