Как преобразовать файл tenorflow .pb в .bytes?

Question

Я пытаюсь преобразовать пример с тензорным потоком Android, предоставленный в gitub tenorflow, в проект 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, и в большинстве случаев это будет работать просто отлично.Посмотрите мой TF Classify пример для Unity.