TensorFlow Serving 02 serving_basic
来源:互联网 发布:xbox360手柄驱动mac版 编辑:程序博客网 时间:2024/06/05 08:08
服务于TensorFlow模型
本教程将向您展示如何使用TensorFlow服务组件导出已训练的TensorFlow模型,并使用标准tensorflow_model_server进行服务。 如果您已经熟悉TensorFlow服务,并且想了解更多关于服务器内部部件的工作原理,请参阅TensorFlow服务高级教程。
本教程使用TensorFlow教程中介绍的简单的Softmax回归模型,用于手写图像(MNIST数据)分类。 如果您不知道TensorFlow或MNIST是什么,请参阅MNIST for ML初学者教程。
本教程的代码由两部分组成:
* 一个Python文件,mnist_saved_model.py,用于训练和导出模型。
* 一个C ++文件main.cc,它是标准的TensorFlow模型服务器,用于发现新的导出模型并运行gRPC服务来提供服务。
开始之前,请完成先决条件。
训练和导出TensorFlow模型
您可以在mnist_saved_model.py中看到,训练与MNIST For ML初学者教程中的相同。 TensorFlow图在TensorFlow会话中启动,输入张量(图像)为x,输出张量(Softmax分数)为y。
然后我们使用TensorFlow的SavedModelBuilder模块来导出模型。 SavedModelBuilder将经过训练的模型的“快照”保存到可靠的存储器中,以便稍后加载进行推理。
有关SavedModel格式的详细信息,请参阅SavedModel README.md上的文档。
从mnist_saved_model.py,以下是一个简短的代码片段,用于说明将模型保存到磁盘的一般过程。
from tensorflow.python.saved_model import builder as saved_model_builder...export_path_base = sys.argv[-1]export_path = os.path.join( compat.as_bytes(export_path_base), compat.as_bytes(str(FLAGS.model_version)))print 'Exporting trained model to', export_pathbuilder = saved_model_builder.SavedModelBuilder(export_path)builder.add_meta_graph_and_variables( sess, [tag_constants.SERVING], signature_def_map={ 'predict_images': prediction_signature, signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY: classification_signature, }, legacy_init_op=legacy_init_op)builder.save()
- TensorFlow Serving 02 serving_basic
- Centos7部署Tensorflow Serving
- 安装tensorflow serving总结
- TensorFlow Serving 00 前言
- TensorFlow Serving 01 安装
- Tensorflow Serving不支持py_func
- Docker使用TensorFlow Serving
- tensorflow serving for centos 7.2
- TensorFlow Serving,GPU版安装
- 谷歌发布 TensorFlow Serving
- tensorflow serving slim配置流程
- TensorFlow Serving 中的最新创新
- TensorFlow Serving 中的最新创新
- TensorFlow学习笔记13----TensorFlow Serving
- 利用 Serving 搭建 TensorFlow 私有云服务
- Tensorflow Serving 模型部署和服务
- Tensorflow serving(tf_serving) 安装测试说明
- docker + tensorflow serving 踩坑记录
- 初次试探
- 判断回文数
- 与秦兽有关的日子
- 方法的重载与重写的关系(区别);static关键字;单例设计模式;继承;
- Java开源-astar:A 星算法(转)
- TensorFlow Serving 02 serving_basic
- 网易云课堂Java进阶学习笔记系列04 -- 第8周 控制反转与MVC模式
- RpcUI4D
- Linux内核页回收swappiness参数确切含义
- Mybatis(一)—实现对数据库的增删改查操作
- 针对架构设计的几个痛点,我总结出的架构原则和模式
- 什么情况下,发送端的send函数成功, 但发送端抓不到对应的网络包?------深入理解send函数
- 如何判断变量是数组还是对象
- 自主编程实现栈和链式栈