Блог

В настоящее время я пытаюсь заморозить файл saved_model.pb в замороженный график вывода, чтобы использовать его для обнаружения объектов tensorflow. Для этого я использую freeze_graph.py скрипт с github tensorflow. Команда, которую я выполняю в командной строке, выглядит следующим образом:

 python freeze_graph.py 
--input_graph=./TFChess/data/models/model_yolo5_v1/saved_model/keras_metadata.pb 
--input_binary=true 
--input_checkpoint=./TFChess/data/models/model_yolo5_v1/checkpoint/best.pt
--output_graph=./TFChess/data/models/model_yolo5_v1/frozen_graph.pb
 

После выполнения этой команды я всегда получаю это сообщение об ошибке:

 Traceback (most recent call last):
  File "C:tensorflowmodelsresearchobject_detectionfreeze_graph.py", line 487, in <module>
    run_main()
  File "C:tensorflowmodelsresearchobject_detectionfreeze_graph.py", line 483, in run_main
    app.run(main=my_main, argv=[sys.argv[0]]   unparsed)
  File "C:UsersNAMEanaconda3envstensorflowlibsite-packagesabslapp.py", line 312, in run
    _run_main(main, args)
  File "C:UsersNAMEanaconda3envstensorflowlibsite-packagesabslapp.py", line 258, in _run_main
    sys.exit(main(argv))
  File "C:tensorflowmodelsresearchobject_detectionfreeze_graph.py", line 482, in <lambda>
    my_main = lambda unused_args: main(unused_args, flags)
  File "C:tensorflowmodelsresearchobject_detectionfreeze_graph.py", line 368, in main
    freeze_graph(flags.input_graph, flags.input_saver, flags.input_binary,
  File "C:tensorflowmodelsresearchobject_detectionfreeze_graph.py", line 334, in freeze_graph
    input_graph_def = _parse_input_graph_proto(input_graph, input_binary)
  File "C:tensorflowmodelsresearchobject_detectionfreeze_graph.py", line 247, in _parse_input_graph_proto
    input_graph_def.ParseFromString(f.read())
google.protobuf.message.DecodeError: Error parsing message