如何在运行 optimize_for_inference.py 时修复“google.protobuf.message.DecodeError：解析消息时出错”答案

【问题标题】：How to fix 'google.protobuf.message.DecodeError: Error parsing message' while running optimize_for_inference.py如何在运行 optimize_for_inference.py 时修复“google.protobuf.message.DecodeError：解析消息时出错”
【发布时间】：2019-01-30 12:20:43
【问题描述】：

我正在对我的自定义数据集重新训练 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

我使用的是 tensorflow 1.4.0 版

【问题讨论】：

标签： android python tensorflow image-segmentation

【解决方案1】：

在将值传递给参数时，我删除了“=”前后的空格。现在可以正常使用了

【讨论】：