如何将张量流 .pb 文件转换为 .bytes?

Question

我正在尝试将tensorflow github中提供的android tensorflow示例转换为Unity项目。我有一个 .pb 文件用于ssd_mobilenet_v1_android_export。但是要在 Unity 中使用张量流模型，您必须拥有 .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，在大多数情况下，这将正常工作。查看我的 Unity 的 TF 分类示例。