gRPC—发送。Protobuf中从Java客户端到Python服务器的png图像

Question

我正在尝试使用gRPC和ProtoBuf将图像从Java客户端发送到Python服务器。我将图像编码为ByteString，并使用blockingStub发送它。Python服务器接收ProtoBuf，但Java ByteString现在是Python str。我不知道如何从str恢复图像。

当我调用服务器时：

request.ParseFromString(request.png_encoded)  # Throws error

它提出：

回溯（最近一次调用）：

文件“/usr/local/lib/python2.7/dist-packages/grpc/server.py”，第364行，在调用行为中

返回行为（参数、上下文），True

进程请求中的文件“process_server.py”，第57行。ParseFromString（request.png_encoded）

ParseFromString self中的文件“/usr/local/lib/python2.7/dist-packages/google/protobuf/message.py”，第185行。MergeFromString（序列化）

文件“/usr/local/lib/python2.7/dist-packages/google/protobuf/internal/python\u message.py”，第1082行，在MergeFromString中，如果self_InternalParse（序列化，0，长度）！=长度：

文件“/usr/local/lib/python2.7/dist packages/google/protobuf/internal/python\u message.py”，第1108行，位于InternalParse new\u pos=local\u SkipField（缓冲区、new\u pos、end、tag\u字节）

文件“/usr/local/lib/html" target="_blank">python2.7/dist-packages/google/protobuf/internal/decoder.py”，第850行，在SkipField中返回WIRETYPE\u TO\u SKIPPER[wire\u-type]（缓冲区，位置，结束）

文件“/usr/local/lib/python2.7/dist-packages/google/protobuf/internal/decoder.py”，第820行，位于RaiseInvalidWireType中

raise \u DecodeError（'标记的导线类型无效'）

DecodeError：标签有无效的线类型。

Java客户端：

public void run_process(String imPath) {
    logger.info("Will try to retrieve boxes with image " + imPath + " ...");
    File path = new File(imPath);
    byte[] imageInByte = null;
    ByteString byteIm = null;
    try {
        BufferedImage bufferedIm = null;
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        bufferedIm = ImageIO.read(path);
        ImageIO.write(bufferedIm, "png", out);
        out.flush();
        imageInByte = out.toByteArray();
        out.close();
        ByteBuffer buf = ByteBuffer.wrap(imageInByte);
        byteIm = ByteString.copyFrom(buf);
    } catch (IOException e) {
        e.printStackTrace();
    }

    private final ManagedChannel channel;
    channel = ManagedChannelBuilder.forAddress("localhost", 50051).usePlaintext(true).build();
    private final ProcessServiceGrpc.ProcessServiceBlockingStub blockingStub;
    blockingStub = ProcessServiceGrpc.newBlockingStub(channel);

    ProcessRequest request = ProcessRequest.newBuilder().setPngEncoded(byteIm).build();
    ProcessResponse response;
    try {
        response = blockingStub.process(request);
    } catch (StatusRuntimeException e) {
        logger.log(Level.WARNING, "RPC failed: {0}", e.getStatus());
        return;
    }
    logger.info("Boxes retrieved: " + response.toString());
}

Python服务器：

def Process(self, request, context):
    print(type(request.png_encoded))  # prints <type 'str'>
    request.ParseFromString(request.png_encoded)  # Throws error

    # Return empty response for now
    process_response = inference_pb2.ProcessResponse()
    return process_response

def serve():
    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
    inference_pb2.add_ProcessServiceServicer_to_server(
        ProcessServiceServicer(), server)
    print "Starting Process Server"
    server.add_insecure_port('localhost:50051')
    server.start()
    try:
        while True:
            time.sleep(_ONE_DAY_IN_SECONDS)
    except KeyboardInterrupt:
        server.stop(0)

if __name__ == '__main__':
    serve()

我曾尝试在谷歌上搜索“用Python解码Java Protobuf”和“用Python测试环解码Java Protobuf”，但没有成功。提前感谢！

Answer 1

Java中的ByteString就是一个不可变的字节[]。Python 2的等价物是str。它是二进制表示。如果要将映像保存到磁盘，请执行以下操作：

with open('out.png', 'wb') as file:
    file.write(request.png_encoded)