对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连接MySQL数据库实例分析
May 12 Python
Python文档生成工具pydoc使用介绍
Jun 02 Python
Python自动化测试ConfigParser模块读写配置文件
Aug 15 Python
Python实现简单的获取图片爬虫功能示例
Jul 12 Python
在python中实现调用可执行文件.exe的3种方法
Jul 07 Python
python利用wx实现界面按钮和按钮监听和字体改变的方法
Jul 17 Python
python程序 创建多线程过程详解
Sep 23 Python
python 经典数字滤波实例
Dec 16 Python
python中matplotlib实现随鼠标滑动自动标注代码
Apr 23 Python
Python新手如何理解循环加载模块
May 29 Python
tensorflow图像裁剪进行数据增强操作
Jun 30 Python
PySwarms(Python粒子群优化工具包)的使用:GlobalBestPSO例子解析
Apr 05 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 mvc开发模式的感想
2011/06/28 PHP
php计算年龄精准到年月日
2015/11/17 PHP
PHP实现数组和对象的相互转换操作示例
2019/03/20 PHP
jQuery实现的Div窗口震动特效
2014/06/09 Javascript
javascript中bind函数的作用实例介绍
2014/09/28 Javascript
JavaScript使用循环和分割来替换和删除元素实例
2014/10/13 Javascript
jquery.mobile 共同布局遇到的问题小结
2015/02/10 Javascript
javascript实现数独解法
2015/03/14 Javascript
Jquery 垂直多级手风琴菜单附源码下载
2015/11/17 Javascript
基于JQuery实现图片轮播效果(焦点图)
2016/02/02 Javascript
jQuery实现响应鼠标事件的图片透明效果【附demo源码下载】
2016/06/16 Javascript
jQuery实现邮箱下拉列表自动补全功能
2016/09/08 Javascript
完美解决jQuery的hover事件在IE中不停闪动的问题
2017/02/10 Javascript
对 Vue-Router 进行单元测试的方法
2018/11/05 Javascript
vuex的module模块用法示例
2018/11/12 Javascript
浅谈ng-zorro使用心得
2018/12/03 Javascript
JS使用数组实现的队列功能示例
2019/03/04 Javascript
Vue+element 解决浏览器自动填充记住的账号密码问题
2019/06/11 Javascript
js/jQuery实现全选效果
2019/06/17 jQuery
Vue scoped及deep使用方法解析
2020/08/01 Javascript
[01:51]开启你的城市传奇 完美世界城市挑战赛开始报名
2018/10/09 DOTA
安装Python和pygame及相应的环境变量配置(图文教程)
2017/06/04 Python
Python实现时钟显示效果思路详解
2018/04/11 Python
python字符串下标与切片及使用方法
2020/02/13 Python
如何安装并在pycharm使用selenium的方法
2020/04/30 Python
利用Python将图片中扭曲矩形的复原
2020/09/07 Python
HTML5进度条特效
2014/12/18 HTML / CSS
详解通过HTML5 Canvas实现图片的平移及旋转变化的方法
2016/03/22 HTML / CSS
美国鲜花递送:UrbanStems
2021/01/04 全球购物
新三好学生主要事迹
2014/01/23 职场文书
粗加工管理制度
2014/02/04 职场文书
《蜗牛》教学反思
2014/02/18 职场文书
小学科学教学计划
2015/01/21 职场文书
教师调动申请报告
2015/05/18 职场文书
学习十八大的感悟
2015/08/11 职场文书
在HTML5 localStorage中存储对象的示例代码
2021/04/21 Javascript