Question

我正在尝试将tensorflow github中提供的android tensorflow示例转换为Unity项目。我有一个ssd_mobilenet_v1_android_export的.pb文件。但是要在Unity中使用tensorflow模型，您必须使用.bytes格式的模型。我无法弄清楚如何将.pb文件转换为.bytes。我打算使用这段代码但是我没有这个图的任何检查点，只有.pb文件。

from tensorflow.python.tools import freeze_graph

freeze_graph.freeze_graph(input_graph = model_path +'/raw_graph_def.pb',
              input_binary = True,
              input_checkpoint = last_checkpoint,
              output_node_names = "action",
              output_graph = model_path +'/your_name_graph.bytes' ,
              clear_devices = True, initializer_nodes = "",input_saver = "",
              restore_op_name = "save/restore_all", filename_tensor_name = "save/Const:0")

有一种简单的方法可以进行此转换吗？或者为这个模型获取检查点的简单方法？看起来这应该是显而易见的，但我无法弄明白。感谢。

Answer 1

您可以将扩展程序从.pb切换到.bytes，对于大多数情况，这样可以正常使用。查看我的TF Classify example了解Unity。

如何将tensorflow .pb文件转换为.bytes？

1 个答案: