Tensorflow 模型保存与恢复（2）使用SavedModel

最新推荐文章于 2023-07-04 14:06:24 发布
最新推荐文章于 2023-07-04 14:06:24 发布
阅读量9.4k 收藏 14
点赞数
分类专栏： TensorFlow 文章标签： tensorflow savedmodel 模型保存恢复 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/JerryZhang__/article/details/85058005
版权
使用SavedModel 保存和恢复模型上一篇介绍了使用tf.train.Saver()保存模型checkpoint的方法，本篇介绍使用SavedModel进行模型的保存与恢复。SavedModel 是一种跨语言的序列化格式（protobuf），可以保存和加载模型变量、图和图的元数据，适用于将训练得到的模型保存用于生产环境中的预测过程。由于跨语言的特性，应用时，可以使用一种语言保存模型，如训练...
摘要由CSDN通过智能技术生成

使用SavedModel 保存和恢复模型

本篇介绍使用SavedModel进行模型的保存与恢复。

其他相关：
Tensorflow 模型保存与恢复（1）使用tf.train.Saver()
Tensorflow 模型保存与恢复（3）保存模型到单个文件中

SavedModel 是一种跨语言的序列化格式（protobuf），可以保存和加载模型变量、图和图的元数据，适用于将训练得到的模型保存用于生产环境中的预测过程。由于跨语言的特性，应用时，可以使用一种语言保存模型，如训练时使用Python代码保存模型；使用另一种语言恢复模型，如使用C++代码恢复模型，进行前向推理，提高效率。
·SavedModel·可以为保存的模型添加签名·signature·，用于保存指定输入输出的graph, 另外可以为模型中的输入输出tensor指定别名，这样子使用模型的时候就不必关心训练阶段模型的输入输出tensor具体的name是什么，讲模型训练和部署解耦，更加方便。

1.保存模型

使用SavedModel保存模型时使用tf.saved_model.builder.SavedModelBuilder来实现，实例化一个builder,传入模型保存的路径：

最低0.47元/天 解锁文章
JerryZhang__
关注
  • 0
    点赞
  • 踩
  • 14
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
专栏目录
tensorflow2.0保存和恢复模型3种方法
09-18
TensorFlow的SavedModel API提供了一种更通用的保存和恢复方式，它不仅包含模型的权重，还包含模型的签名和元数据。这使得模型可以在不同的计算平台，如TensorFlow Serving或TensorFlow Lite中使用。 ```python # ...
tf.saved_model.load的用法
martinkeith的博客
09-16 4326
tensorflow2.3.0 Load a SavedModel from export_dir. tf.saved_model.load( export_dir, tags=None, options=None ) 参考链接：https://www.tensorflow.org/api_docs/python/tf/saved_model/load
tensorflow模型保存（三）——tensorflow1.x版本的savedmodel格式的模型保存与加载
MIss-Y的博客
05-06 1万+
前言：前面的两篇文章已经分别介绍了高层API与低层API关于各种模型的保存，比如在tensorflow2.x中如何保存checkpoint与savedmodel，在tensorflow1.x中如何保存checkpoint，本文着重介绍在tensorflow1.x版本中怎么保存savedmodel。 要想恢复模型并能够正确的加载，最核心的就在于要给模型的输入与输出起一...
TensorFlow - 保存和恢复
既然选择了远方 便只顾风雨兼程 - 永强
08-06 1513
TensorFlow - 保存和恢复 https://tensorflow.google.cn/guide/saved_model TensorFlow 指南 - TensorFlow 工作原理 https://tensorflow.google.cn/guide tf.train.Saver 类提供了保存和恢复模型的方法。通过 tf.saved_model.simple_save 函数可轻松地保存适合投入使用的模型。Estimator 会自动保存和恢复 model_dir 中的变量。 1. 保存和恢复变量
TensorFlow2.0 —— 模型保存与加载
然后就去远行
04-04 1万+
目录1、Keras版本模型保存与加载2、自定义版本模型保存与加载3、总结 1、Keras版本模型保存与加载 保存模型权重（model.save_weights） 保存HDF5文件（model.save） 保存pb文件（tf.saved_model） tf.saved_model和model.save的区别在于，tf.saved_model格式的模型可以直接用来预测，但是tf.saved_model没有保存优化器配置，而model.save保存了优化器配置，所以整体更大。 保存模型权重方法仅仅保存了模型中
Tensorflow：模型保存和服务
皮皮blog
07-13 3054
tensorflow模型保存和使用 TensorFlow是通过构造Graph的方式进行深度学习，任何操作(如卷积、池化等)都需要operator，保存和恢复操作也不例外。在tf.train.Saver()类初始化时，用于保存和恢复的save和restore operator会被加入Graph，所以类初始化操作应在搭建Graph时完成。TensorFlow会将变量保存在二进制checkpoint文...
tf.saved_model.load（）出错
dream_to_dream的博客
06-21 1764
有时我们保存tensorflow2.x模型时，采用了tensorflow的统一模型保存格式， tf.saved_model.save(model, "保存的目标文件夹名称") 当我们di
TensorFlow模型保存和提取的方法
09-20
以下是对TensorFlow模型保存和提取的详细解释： 首先，TensorFlow提供了`tf.train.Saver`类来实现模型的保存和恢复。在保存模型时，我们创建一个`Saver`对象，并调用其`save`方法，如`saver.save(sess,"Model/model...
TensorFlow实现保存训练模型为pd文件并恢复
12-20
在这种情况下，可以考虑使用`SavedModel` API，它提供了更高级的保存和恢复功能，包括保存模型的签名和版本信息，使得模型可以在TensorFlow Serving等环境中直接部署。 总的来说，TensorFlow提供了多种方式来保存和...
如何解决tensorflow恢复模型的特定值时出错
12-20
在TensorFlow 2.x中，推荐使用`tf.saved_model.save()`和`tf.keras.models.load_model()`进行模型的保存和加载，它们更加简洁且易于使用。对于TensorFlow 1.x，`tf.train.Saver()`是常用的方式。 在某些场景下，...
python模型保存save_Tensorflow SavedModel 模型的保存和加载
weixin_39928017的博客
12-11 1293
前言最近参加了天池上的Apache Flink极客挑战赛——垃圾图片分类比赛，里面涉及到了Java调用tensorflow的SavedModel格式的模型进行预测，于是专门对此内容进行了调研。这里记录了SavedModel模型的优势，结构以及保存和加载的方法。SavedModel的优势Tensorflow训练的模型可以保存为ckpt格式，但是这种格式的模型文件在跨语言方面不是很灵活。而SaveMo...
TensorFlow模型的保存及模型的应用
QQ1131221088的博客
10-31 457
tensorflow的模型保存形式？ 1.ckpt格式 就是通过如下几个函数实现的tensorflow模型保存的模型，是ckpt格式的模型。 saver = tf.train.Saver() ... saver.save(sess, saveFile) 就可以保存出如下文件： checkpoint model-450.data-00000-of-00001 model-450.index mod...
TensorFlow学习记录：saved_model模块的用法
weixin_41137655的博客
08-14 9649
TensorFlow中的saved_model模块用于生成冻结图文件，并且saved_model模块封装了平常用的Saver类。与Saver类不同的是，saved_model模块生成的模型文件集成了打标签的操作，可以更方便地部署在生产环境中。 关于为什么要用saved_model模块，这篇文章讲得挺好的。请点击这里 一个saved_model对象可以存储一个或多个MetaGraphDef。那什么时...
TensorFlow - save model
Zhou_Dao的博客
12-27 436
There are some examples to show how to save trained models example 1: this code can save the model which achieves max acc; but it cannot train model based on the pre-trained model; Besides, it e...
TensorFlow中的模型保存和加载方法是什么？
最新发布
m0_74693860的博客
07-04 1033
在TensorFlow中，模型的保存和加载是非常重要的功能，它使得我们可以在训练后保存模型并在需要时重新加载模型进行预测或继续训练。TensorFlow提供了几种方法来保存和加载模型，包括Checkpoint和SavedModel。下面我将介绍这两种方法的基本原理和使用方法。
tf.saved_model.save模型导出、TensorFlow Serving模型部署、TensorBoard中的HParams 超参数调优
あずにゃん梓喵的博客
08-04 4256
日萌社 人工智能AI：Keras PyTorch MXNet TensorFlow PaddlePaddle 深度学习实战（不定时更新） 4.11 综合案例：模型导出与部署 学习目标 目标 掌握TensorFlow模型的导出(saved_model格式) 掌握Tensorflow模型的部署 掌握TensorFlow模型的客户端调用 掌握TensorFlow模型的超参...
TensorFlow saved_model 模块
热门推荐
thriving_fcl的博客
07-17 4万+
saved_model模块主要用于TensorFlow Serving。TF Serving是一个将训练好的模型部署至生产环境的系统，主要的优点在于可以保持Server端与API不变的情况下，部署新的算法或进行试验，同时还有很高的性能。保持Server端与API不变有什么好处呢？有很多好处，我只从我体会的一个方面举例子说明一下，比如我们需要部署一个文本分类模型
C# WPF调用Python-Tensorflow2深度学习模型
LemonQ的博客
09-06 5663
C# WPF调用python-tensorflow2深度学习模型一 环境介绍二 tensorflow模型的训练和生成1 加载数据训练模型2 h5文件转pb二 C#加载模型并预测1 vs2017环境搭建2 调用模型三 最终效果 python在研究深度学习人工智能领域十分强大，但在工业项目开发中仍常常使用C#和C++来做软件，C++有Caffe深度学习框架，但C#尚且没有成熟的深度学习框架（有个tensroflow.net尚在开发中，有兴趣可以去研究研究）。现在实验室项目开发又要用C#，经过实践最终决定在C#端
【TensorFlow2.x开发—基础】 模型保存、加载、使用
黎国溥
04-27 3761
本文主要介绍在TensorFlow2 中使用Keras API保存整个模型，以及如果使用保存好的模型。保存整个模型时，有两种格式可以实现，分别是SaveModel和HDF5；在TF2.x中模型使用SavedModel格式。 文章分为简约版、实践版、代码版，首先从简约版认识基本流程、概率、命令等；再到实践版查看每一步的代码调试结果；最后通过代码版，可以自己尝试实践。
tensorflow2保存模型格式
04-01
1. SavedModel 格式：SavedModel 是 TensorFlow 2 推荐的模型保存格式。它是一个文件夹，包含了模型的计算图、变量、签名等信息。保存后可以方便地加载到 TensorFlow Serving 中进行部署和预测。使用 tf.saved_model...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值