对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 当前全局变量和入口参数的所有属性
Jul 01 Python
python3简单实现微信爬虫
Apr 09 Python
python递归计算N!的方法
May 05 Python
Python 26进制计算实现方法
May 28 Python
基于Python的XSS测试工具XSStrike使用方法
Jul 29 Python
Python Nose框架编写测试用例方法
Oct 26 Python
简单了解OpenCV是个什么东西
Nov 10 Python
利用Python找出序列中出现最多的元素示例代码
Dec 08 Python
python 将字符串转换成字典dict的各种方式总结
Mar 23 Python
python 调用钉钉机器人的方法
Feb 20 Python
Python猫眼电影最近上映的电影票房信息
Sep 18 Python
python+pyhyper实现识别图片中的车牌号思路详解
Dec 24 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
2020年4月放送!《Princess Connect Re:Dive》制作组 & 角色声优公开!
2020/03/06 日漫
PHP 实现多服务器共享 SESSION 数据
2009/08/15 PHP
PHP 多维数组排序(usort,uasort)
2010/06/30 PHP
php生成二维码不保存服务器还有下载功能的实现代码
2018/08/09 PHP
Laravel5.5 数据库迁移:创建表与修改表示例
2019/10/23 PHP
PHP连接SQL server数据库测试脚本运行实例
2020/08/24 PHP
php使用event扩展的io复用测试的示例
2020/10/20 PHP
xmlhttp缓存清除的2种解决方法
2013/12/13 Javascript
Area 区域实现post提交数据的js写法
2014/04/22 Javascript
基于Jquery和html5实现炫酷的3D焦点图动画
2016/03/02 Javascript
jQuery实现定位滚动条位置
2016/08/05 Javascript
JS原型链 详解及示例代码
2016/09/06 Javascript
Vue.js每天必学之组件与组件间的通信
2016/09/08 Javascript
jQuery中的一些小技巧
2017/01/18 Javascript
ES6新特性之Object的变化分析
2017/03/31 Javascript
基于BootStrap实现简洁注册界面
2017/07/20 Javascript
vue路由嵌套的SPA实现步骤
2017/11/06 Javascript
jQuery实现的鼠标响应缓冲动画效果示例
2018/02/13 jQuery
在Django中限制已登录用户的访问的方法
2015/07/23 Python
python脚本实现数据导出excel格式的简单方法(推荐)
2016/12/30 Python
python程序 创建多线程过程详解
2019/09/23 Python
基于pandas中expand的作用详解
2019/12/17 Python
python利用tkinter实现图片格式转换的示例
2020/09/28 Python
CSS3基础(RGBa、text-shadow、box-shadow、border-radius)
2012/11/13 HTML / CSS
canvas实现飞机打怪兽射击小游戏的示例代码
2018/07/09 HTML / CSS
Auchan Direct波兰:欧尚在线杂货店
2016/10/19 全球购物
Linux文件操作命令都有哪些
2016/07/23 面试题
研究生毕业鉴定
2014/01/29 职场文书
学习优秀党务工作者先进事迹材料思想报告
2014/09/17 职场文书
会计试用期自我评价
2014/09/19 职场文书
作风建设剖析材料
2014/10/06 职场文书
乱丢垃圾袋检讨书
2014/10/08 职场文书
汽车销售员岗位职责
2015/04/11 职场文书
2015年行政助理工作总结
2015/04/30 职场文书
2015年化工厂工作总结
2015/05/04 职场文书
职位证明模板
2015/06/23 职场文书