对tensorflow 的模型保存和调用实例讲解


Posted in Python onJuly 28, 2018

我们通常采用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 的模型保存和调用实例讲解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中多线程及程序锁浅析
Jan 21 Python
Python基于回溯法子集树模板解决选排问题示例
Sep 07 Python
Python SqlAlchemy动态添加数据表字段实例解析
Feb 07 Python
python安装教程
Feb 28 Python
python list是否包含另一个list所有元素的实例
May 04 Python
python实现图像检索的三种(直方图/OpenCV/哈希法)
Aug 08 Python
Django之使用celery和NGINX生成静态页面实现性能优化
Oct 08 Python
Python实现平行坐标图的绘制(plotly)方式
Nov 22 Python
python 实现一个反向单位矩阵示例
Nov 29 Python
解决Python列表字符不区分大小写的问题
Dec 19 Python
pytorch之Resize()函数具体使用详解
Feb 27 Python
django中嵌套的try-except实例
May 21 Python
Python Socket编程之多线程聊天室
Jul 28 #Python
python实现点对点聊天程序
Jul 28 #Python
基于python实现聊天室程序
Jul 27 #Python
Python中return self的用法详解
Jul 27 #Python
TensorFlow打印tensor值的实现方法
Jul 27 #Python
解决tensorflow测试模型时NotFoundError错误的问题
Jul 27 #Python
Tensorflow 同时载入多个模型的实例讲解
Jul 27 #Python
You might like
PHP 反向排序和随机排序代码
2010/06/30 PHP
PHP数字和字符串ID互转函数(类似优酷ID)
2014/06/30 PHP
PHP排序算法类实例
2015/06/17 PHP
PHP实现支付宝即时到账功能
2016/12/21 PHP
浅谈PHP发送HTTP请求的几种方式
2017/07/25 PHP
mysqli扩展无法在PHP7下升级问题的解决
2019/09/10 PHP
零基础学JavaScript最新动画教程+iso光盘下载
2008/01/22 Javascript
js 日期转换成中文格式的函数
2009/07/07 Javascript
JavaScript作用域与作用域链深入解析
2013/12/06 Javascript
javascript实现表格排序 编辑 拖拽 缩放
2015/01/02 Javascript
怎么通过onclick事件获取js函数返回值(代码少)
2015/07/28 Javascript
JavaScript的jQuery库中function的存在和参数问题
2015/08/13 Javascript
浅析nodejs实现Websocket的数据接收与发送
2015/11/19 NodeJs
javascript实现Email邮件显示与删除功能
2015/11/21 Javascript
jQuery简单实现上下,左右滑动的方法
2016/06/01 Javascript
javascript设计模式之Adapter模式【适配器模式】实现方法示例
2017/01/13 Javascript
利用js定义一个导航条菜单
2017/03/14 Javascript
JS实现经典的中国地区三级联动下拉菜单功能实例【测试可用】
2017/06/06 Javascript
vue中Element-ui 输入银行账号每四位加一个空格的实现代码
2018/09/14 Javascript
vue-router两种模式区别及使用注意事项详解
2019/08/01 Javascript
[04:50]DOTA2亚洲邀请赛小组赛第四日 TOP10精彩集锦
2015/02/02 DOTA
在漏洞利用Python代码真的很爽
2007/08/26 Python
Python urlopen()函数 示例分享
2014/06/12 Python
itchat接口使用示例
2017/10/23 Python
Python3 replace()函数使用方法
2018/03/19 Python
Python库skimage绘制二值图像代码实例
2020/04/10 Python
北美领先的智能产品购物网站:Wellbots
2018/06/11 全球购物
俄罗斯药房连锁店:ASNA
2020/06/20 全球购物
应届大专生求职信
2014/06/26 职场文书
民主评议政风行风活动心得体会
2014/10/29 职场文书
班主任自我评价范文
2015/03/11 职场文书
义卖募捐活动总结
2015/05/09 职场文书
小公司融资,商业计划书的8切记
2019/07/15 职场文书
承诺书应该怎么写?
2019/09/10 职场文书
python基于opencv批量生成验证码的示例
2021/04/28 Python
详解Python内置模块Collections
2022/03/22 Python