无法将重复的 protobuf 对象保存到文件 (Python)

Question

我是 protobuf 的新手，所以我不知道如何正确地提出问题。

无论如何，我正在使用这个Model Config proto file. I converted it into python using this command protoc -I=. --python_out=. ./model_server_config.proto from Protocol Buffer page。现在我有一些可以导入和处理的 python 文件。我的 objective 是创建一个文件（用于运行具有多个模型的 TensorFlow 模型服务器），它应该如下所示：

model_config_list: {
 config: {
    name: "name1",
    base_path: "path1",
    model_platform: "tensorflow"
  },
  config: {
    name: "name2",
    base_path: "path2",
    model_platform: "tensorflow"
  },
  config: {
    name: "name3",
    base_path: "path3",
    model_platform: "tensorflow"
  },
}

现在使用编译的 python 包，我制作了一个 protobuf 对象，当我打印出来时它看起来像这样：

model_config_list {
  config {
    name: "name1"
    base_path: "path1"
    model_platform: "tensorflow"
  }
  config {
    name: "name2"
    base_path: "path2"
    model_platform: "tensorflow"
  }
  config {
    name: "name3"
    base_path: "path3"
    model_platform: "tensorflow"
  }
}

但是在使用 objectname.SerializeToString() 序列化对象时，我得到一个奇怪的输出：

b'\n\x94\x01\n \n\x04name1\x12\x0cpath1"\ntensorflow\n7\n\x08name2\x12\x1fpath2"\ntensorflow\n7\n\x08name3\x12\x1fpath3"\ntensorflow'

我尝试将其转换为 Json，也使用 python 的 protobuf，如下所示：

from google.protobuf.json_format import MessageToJson
MessageToJson(objectname)

这给了我这样的结果：

{
  "modelConfigList": {
    "config": [
      {
        "name": "name1",
        "basePath": "path1",
        "modelPlatform": "tensorflow"
      },
      {
        "name": "name2",
        "basePath": "path2",
        "modelPlatform": "tensorflow"
      },
      {
        "name": "name3",
        "basePath": "path3",
        "modelPlatform": "tensorflow"
      }
    ]
  }
}

所有对象都在一个列表中，每个对象都是字符串，这对于 TensorFlow 模型服务器配置是不可接受的。

关于如何将其正确写入文件的任何想法？还是我错误地创建了整个对象？欢迎任何帮助，提前致谢。

Answer 1

我不知道什么系统会读取你的文件，所以我不能说你应该如何将它写入文件。这实际上取决于模型服务器期望如何读取它。

也就是说，我看不出您创建消息的方式或您展示的任何序列化方法有任何问题。

print方法展示了一个“文本格式”proto，有利于调试，有时用于存储配置文件。它不是很紧凑（字段名称存在于文件中）并且不具有二进制表示的所有向后和向前兼容的功能。它实际上在功能上与您所说的“应该看起来像”相同：冒号和逗号实际上是可选的。
SerializeToString()方法使用二进制serialization format。这可以说是 Protocol Buffers 的构建目的。它是一种紧凑的表示形式，并提供向后和向前的兼容性，但人类可读性不高。
顾名思义，json_format 模块提供消息的 JSON 表示。如果您正在与之交互的系统需要 JSON，那就太好了，但这并不常见。

附录：google.protobuf.text_format 模块不使用 print()，而是具有更适合以编程方式使用文本格式的实用程序。要写入文件，您可以使用：

from google.protobuf import text_format
(...)
with open(file_path, 'w') as output:
  text_format.PrintMessage(my_message, output)

无法将重复的 protobuf 对象保存到文件 (Python)

Trouble saving repeated protobuf object to file (Python)

python

protocol-buffers

proto

tensorflow

protobuf-python