Как исправить «google.protobuf.message.DecodeError: Ошибка при разборе сообщения» при запуске optimize_for_inference.py - PullRequest
Новая
       7

Как исправить «google.protobuf.message.DecodeError: Ошибка при разборе сообщения» при запуске optimize_for_inference.py

0 голосов
/ 30 января 2019

Я переобучаюсь mobilenet_v1_1.0_224 в моем пользовательском наборе данных.Я сгенерировал требуемый файл graph.pb и хочу запустить модель в приложении для Android.Попытался запустить optimize_for_inference.py для создания оптимизированного файла, но получил сообщение об ошибке - google.protobuf.message.DecodeError: Error parsing message

Я проверил установку protobuf, и она присутствует.Я запускаю следующее - 

!python tensorflow/tensorflow/python/tools/optimize_for_inference.py \
--input = model_output/graph_v1.pb \
--output = model_output/optimized_graph_v1.pb \
--input_names = input \
--output_names = MobilenetV1/Predictions/Reshape_1

Вывод: 

Traceback (most recent call last):
  File "tensorflow/tensorflow/python/tools/optimize_for_inference.py", line 165, in <module>
    app.run(main=main, argv=[sys.argv[0]] + unparsed)
  File "/anaconda3/lib/python3.6/site-packages/tensorflow/python/platform/app.py", line 48, in run
    _sys.exit(main(_sys.argv[:1] + flags_passthrough))
  File "tensorflow/tensorflow/python/tools/optimize_for_inference.py", line 83, in main
    input_graph_def.ParseFromString(data)
google.protobuf.message.DecodeError: Error parsing message

Использую версию tenorflow 1.4.0

1 Ответ

0 голосов
/ 31 января 2019

Я удалил пробелы до и после '=' при передаче значений аргументам.Теперь все отлично работает

...