对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记录详细调用堆栈日志的方法
May 05 Python
Python的Twisted框架中使用Deferred对象来管理回调函数
May 25 Python
Selenium鼠标与键盘事件常用操作方法示例
Aug 13 Python
Pycharm2017版本设置启动时默认自动打开项目的方法
Oct 29 Python
详解Python logging调用Logger.info方法的处理过程
Feb 12 Python
python 设置输出图像的像素大小方法
Jul 04 Python
Python生命游戏实现原理及过程解析(附源代码)
Aug 01 Python
使用Matplotlib 绘制精美的数学图形例子
Dec 13 Python
TensorFlow实现checkpoint文件转换为pb文件
Feb 10 Python
python3将变量写入SQL语句的实现方式
Mar 02 Python
python中for in的用法详解
Apr 17 Python
Python PyQt5模块实现窗口GUI界面代码实例
May 12 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
Windows下的PHP5.0详解
2006/11/18 PHP
php地址引用(php地址引用的效率问题)
2012/03/23 PHP
PHP+ajax分页实例简析
2015/12/07 PHP
WordPress中用于更新伪静态规则的PHP代码实例讲解
2015/12/18 PHP
PHP实现Google plus的好友拖拽分组效果
2016/10/21 PHP
在标题栏显示新消息提示,很多公司项目中用到这个方法
2011/11/04 Javascript
多种方法实现JS动态添加事件
2013/11/01 Javascript
利用浏览器全屏api实现js全屏
2014/01/16 Javascript
js实现的点击数量加一可操作数据库
2014/05/09 Javascript
使用jquery.qrcode生成彩色二维码实例
2014/08/08 Javascript
JavaScript获得当前网页来源页面(即上一页)的方法
2015/04/03 Javascript
jQuery背景插件backstretch使用指南
2015/04/21 Javascript
JS实现的相册图片左右滚动完整实例
2016/11/23 Javascript
JavaScript学习笔记--常用的互动方法
2016/12/07 Javascript
NodeJS处理Express中异步错误
2017/03/26 NodeJs
JS正则验证多个邮箱完整实例【邮箱用分号隔开】
2017/04/19 Javascript
基于Vue的SPA动态修改页面title的方法(推荐)
2018/01/02 Javascript
vue使用better-scroll实现下拉刷新、上拉加载
2018/11/23 Javascript
vue实现带过渡效果的下拉菜单功能
2020/02/19 Javascript
详解小程序横屏方案对比
2020/06/28 Javascript
Python基于select实现的socket服务器
2016/04/13 Python
fastcgi文件读取漏洞之python扫描脚本
2017/04/23 Python
python2.6.6如何升级到python2.7.14
2018/04/08 Python
Python 数值区间处理_对interval 库的快速入门详解
2018/11/16 Python
python利用thrift服务读取hbase数据的方法
2018/12/27 Python
Python设计模式之状态模式原理与用法详解
2019/01/15 Python
详解Python3定时器任务代码
2019/09/23 Python
荷兰街头时尚之家:Funkie House
2019/03/18 全球购物
美国摩托车头盔、零件、齿轮及配件商店:Cycle Gear
2019/06/12 全球购物
波兰最大的宠物用品网上商店:FERA.PL
2019/08/11 全球购物
瑞典最大的儿童用品网上商店:pinkorblue.se
2021/03/09 全球购物
Bloomingdale’s阿联酋:选购奢华时尚、美容及更多
2020/09/22 全球购物
英语专业毕业生求职简历的自我评价
2013/10/24 职场文书
调查研究项目计划书
2014/04/29 职场文书
会议营销主持词
2015/07/03 职场文书
教你用python控制安卓手机
2021/05/13 Python