对tensorflow 的模型保存和调用实例讲解
更新时间：2018年07月28日 08:55:33 作者：luoganttcc
今天小编就为大家分享一篇对tensorflow 的模型保存和调用实例讲解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
我们通常采用tensorflow来训练，训练完之后应当保存模型,即保存模型的记忆（权重和偏置），这样就可以来进行人脸识别或语音识别了。
1.模型的保存
# 声明两个变量
v1 = tf.Variable(tf.random_normal([1, 2]), name="v1")
v2 = tf.Variable(tf.random_normal([2, 3]), name="v2")
init_op = tf.global_variables_initializer() # 初始化全部变量
saver = tf.train.Saver() # 声明tf.train.Saver类用于保存模型
with tf.Session() as sess:
sess.run(init_op)
print("v1:", sess.run(v1)) # 打印v1、v2的值一会读取之后对比
print("v2:", sess.run(v2))
#定义保存路径，一定要是绝对路径，且用‘/ '分隔父目录与子目录
saver_path = saver.save(sess, "C:/Users/Administrator/Desktop/tt/model.ckpt") # 将模型保存到save/model.ckpt文件
print("Model saved in file:", saver_path)
2.模型的读取
直接读取模型时，可能会报错，我是用Spyder编译的，可以把Spyder关掉，再重新打开，就可以读取数据了。原因可能是：在模型保存时将变量初始化了。
import tensorflow as tf
# 使用和保存模型代码中一样的方式来声明变量
v1 = tf.Variable(tf.random_normal([1, 2]), name="v1")
v2 = tf.Variable(tf.random_normal([2, 3]), name="v2")
saver = tf.train.Saver() # 声明tf.train.Saver类用于保存模型
with tf.Session() as sess:
saver.restore(sess, "C:/Users/Administrator/Desktop/tt/model.ckpt") # 即将固化到硬盘中的Session从保存路径再读取出来
print("v1:", sess.run(v1)) # 打印v1、v2的值和之前的进行对比
print("v2:", sess.run(v2))
print("Model Restored")
以上这篇对tensorflow 的模型保存和调用实例讲解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Pytorch实现将模型的所有参数的梯度清0，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
Python cookbook(数据结构与算法)将名称映射到序列元素中的方法
这篇文章主要介绍了Python cookbook(数据结构与算法)将名称映射到序列元素中的方法,结合实例形式分析了Python使用collections.namedtuple()进行元组命名相关操作技巧,需要的朋友可以参考下2018-03-03
这篇文章主要介绍了python scrapy框架的日志文件问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-08-08
pandas 是基于NumPy 的一种工具，该工具是为了解决数据分析任务而创建的，这篇文章主要介绍了python中pandas常用命令,需要的朋友可以参考下2022-07-07
生成器（Generators）和迭代器（Iterators）是 Python 中用于处理序列数据的强大工具，本文主要来和大家介绍一下它们的具体使用，方便大家更好的了解它们，需要的可以学习下2022-03-03
在 Rest Framework 中提供了两个 router ， 可以帮助我们快速的实现路由的自动生成，本文通过实例代码给大家介绍drf-router和authenticate认证源码分析，感兴趣的朋友跟随小编一起看看吧2021-07-07
在本篇文章里小编给大家整理的是一篇关于python全栈开发语法的相关总结内容，有兴趣的朋友们可以跟着学习参考下。2020-11-11
下面就使用VLAD表示图像，实现一个小型的图像数据库的检索程序。下面实现需要的功能模块，分步骤给大家介绍的非常详细，对OpenCV图像数据库检索功能感兴趣的朋友跟随小编一起看看吧2021-12-12
今天小编就为大家分享一篇python 读取目录下csv文件并绘制曲线v111的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-07-07
VOC 的标注是xml格式的，而YOLO是.txt格式，所以要实现VOC数据集转YOLO数据集，只能利用代码实现。所以本文为大家介绍了Python中xml转txt的示例代码，需要的可以参考一下2022-03-03
最新评论