因为这可能对其他人有帮助,所以我也会在github ;-)之后回答。我认为您可以尝试这样的事情(使用tensorflow / python / tools中的freeze_graph脚本):
python freeze_graph.py --input_graph=/path/to/graph.pbtxt --input_checkpoint=/path/to/model.ckpt-22480 --input_binary=false --output_graph=/path/to/frozen_graph.pb --output_node_names="the nodes that you want to output e.g. InceptionV3/Predictions/Reshape_1 for Inception V3 "
这里的重要标志是--input_binary = false,因为文件graph.pbtxt是文本格式。我认为它对应于所需的graph.pb,它相当于二进制格式。
关于output_node_names,这确实让我感到困惑,因为我在这部分上仍然存在一些问题,但是您可以在tensorflow中使用summary_graph脚本,该脚本可以将pb或pbtxt用作输入。
问候,
史蒂芬
0
在tensorflow中,从头开始的训练产生了以下6个文件:
我想将它们(或仅需要的文件)转换为一个文件graph.pb ,以便能够将其传输到我的Android应用程序中。
我尝试了脚本
freeze_graph.py
但是它已经需要输入我没有的input.pb文件作为输入。 (我只有前面提到的这6个文件)。如何继续获得一个freeze_graph.pb文件?我看到了几个线程,但没有一个对我有用。