编译 tensorflow-serving-api 生成jar

举报
李玮顾 发表于 2019/05/28 12:02:41 2019/05/28
【摘要】 编译 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


【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。