编译 tensorflow-serving-api 生成jar
1.目的
用Java调用Tfserving服务,需要编译 tensorflow-serving-api 客户端。
2.操作
2.1 环境准备
安装mvn,git,java,protoc;配置JAVA_HOME,将protoc加入PATH。
protobuf下载地址:
https://github.com/protocolbuffers/protobuf/releases
2.2 下载文件
git clone https://github.com/alexxyjiang/tensorflow-serving-api.git
cd tensorflow-serving-api/
git submodule update --init --remote --recursive
2.3 拷贝proto文件
修改initialize.sh文件,将
proto_files=`find . | grep -P '\.proto$' | grep -vP 'host'`
改为
proto_files=`find . | grep -P '\.proto$'
否则不会拷贝 host_compute_metadata.proto 文件。
然后执行
bash initialize.sh
2.4 编译
修改 tensorflow-serving-api/src/main/proto/tensorflow/compiler/tf2xla/host_compute_metadata.proto ,将
option java_outer_classname = "Tf2XlaProtos";
改为
option java_outer_classname = "HostComputeMetadataProtos";
(不改的话,会报错:Tf2XlaProtos.java: Tried to write the same file twice)
执行编译命令:
mvn clean install
完成后会在target目录下看的 tensorflow-serving-api-1.8.0.jar 等编译结果文件。
3 参考资料
https://github.com/alexxyjiang/tensorflow-serving-api
https://www.jianshu.com/p/06c744690250
https://github.com/YuanWenqing/tensorflow4j
https://github.com/tensorflow/tensorflow/issues/25222
- 点赞
- 收藏
- 关注作者
评论(0)