如何为Tensorflow模型服务器创建Java客户端

Question

到目前为止，我已设法导出Tensorflow模型，然后使用Tensorflow模型服务器提供该模型。然后我就可以通过创建一个通过gRPC与Tensorflow服务器通信的客户端来进行预测。

基本上我已经遵循了tutorial

到目前为止，这一切都已在Python中完成，这很好，但我需要能够使用Java调用Tensorflow服务器。换句话说，我需要创建一个Java客户端。

我的基本理解是我需要使用Protoc编译器将.proto文件编译成Java类，但我不知道如何为我的模型生成.proto文件。我已导出模型，但文件采用“.pb”格式，似乎无法编译为Java类。

有没有人有这方面的经验？如果您有，请分享您的解决方案。

非常感谢，

百里

Answer 1

.pb似乎是序列化的protobuf数据。 Protobuf定义是.proto文件，我猜这些文件可以从TensorFlow项目下载，而不是从模型中生成。