你如何在Tensorflow中使用freeze_graph.py?

时间:2018-04-15 02:04:24

标签: python tensorflow machine-learning

我无法在Tensorflow中冻结模型。我想在Unity中使用它,但它们需要冻结图形作为.bytes文件而不是.pb。有没有另一种方法来获取除freeze_graph.py之外的字节文件?我一直收到这个错误:

TypeError:names_to_saveables必须是字符串映射到Tensors / Variables的字符串名称。不是变量:Tensor(" BoxPredictor_0 / BoxEncodingPredictor / biases:0",shape =(12,),dtype = float32)

这就是我目前使用的:

freeze_graph.freeze_graph(input_graph = 'ssd_mobilenet_v1_coco_11_06_2017/frozen_inference_graph.pb',
input_binary = True,
input_checkpoint = 'ssd_mobilenet_v1_coco_11_06_2017/model.ckpt',
output_node_names = "num_detections,detection_boxes,detection_scores,detection_classes",
output_graph = 'BytesModel/bytesModel.bytes',
clear_devices = True, initializer_nodes = "",input_saver = "",
restore_op_name = "save/restore_all", filename_tensor_name = "save/Const:0")

0 个答案:

没有答案