对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下的XSLT API进行web开发的简单教程
Apr 15 Python
利用Python-iGraph如何绘制贴吧/微博的好友关系图详解
Nov 02 Python
基于Django的ModelForm组件(详解)
Dec 07 Python
浅谈Matplotlib简介和pyplot的简单使用——文本标注和箭头
Jan 09 Python
对Python3中的input函数详解
Apr 22 Python
Python使用pydub库对mp3与wav格式进行互转的方法
Jan 10 Python
如何使用Python自动控制windows桌面
Jul 11 Python
基于python进行抽样分布描述及实践详解
Sep 02 Python
解决pycharm 安装numpy失败的问题
Dec 05 Python
python 多维高斯分布数据生成方式
Dec 09 Python
解决pytorch报错:AssertionError: Invalid device id的问题
Jan 10 Python
python软件都是免费的吗
Jun 18 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中截取字符串支持utf-8
2007/01/18 PHP
PHP中include与require使用方法区别详解
2013/10/19 PHP
关于URL最大长度限制的相关资料查证
2014/12/23 PHP
php类的定义与继承用法实例
2015/07/07 PHP
一个PHP实现的轻量级简单爬虫
2015/07/08 PHP
JsEasy简介 JsEasy是什么?与下载
2007/03/07 Javascript
Javascript 同时提交多个Web表单的方法
2009/02/19 Javascript
IE浏览器打印的页眉页脚设置解决方法
2009/12/08 Javascript
JS中confirm,alert,prompt函数区别分析
2011/01/17 Javascript
40个有创意的jQuery图片、内容滑动及弹出插件收藏集之一
2011/12/31 Javascript
JS操作图片(增,删,改) 例子
2013/04/17 Javascript
使用闭包对setTimeout进行简单封装避免出错
2013/07/10 Javascript
Js+php实现异步拖拽上传文件
2015/06/23 Javascript
jQuery动态添加
2016/04/07 Javascript
深入理解$.each和$(selector).each
2016/05/15 Javascript
模仿password输入框的实现代码
2016/06/07 Javascript
JavaScript定时器实现的原理分析
2016/12/06 Javascript
JS简单实现获取元素的封装操作示例
2017/04/07 Javascript
解决百度Echarts图表坐标轴越界的方法
2018/10/17 Javascript
layui 实现自动选择radio单选框(checked)的方法
2019/09/03 Javascript
jQuery实现高度灵活的表单验证功能示例【无UI】
2020/04/30 jQuery
[02:05]2014DOTA2国际邀请赛 BBC外卡赛赛后总结
2014/07/09 DOTA
[00:31]DOTA2上海特级锦标赛 Fnatic战队宣传片
2016/03/04 DOTA
[01:32:50]DOTA2-DPC中国联赛 正赛 DLG vs XG BO3 第一场 1月25日
2021/03/11 DOTA
python实现进程间通信简单实例
2014/07/23 Python
在Python程序中进行文件读取和写入操作的教程
2015/04/28 Python
Python3 加密(hashlib和hmac)模块的实现
2017/11/23 Python
python从内存地址上加载python对象过程详解
2020/01/08 Python
Python装饰器用法与知识点小结
2020/03/09 Python
Django 用户认证Auth组件的使用
2020/11/30 Python
香港太阳眼镜网上商店:SmartBuyGlasses香港
2016/07/22 全球购物
入党自我鉴定范文
2013/10/04 职场文书
计算机考试作弊检讨书1000字
2015/01/01 职场文书
股东大会通知
2015/04/24 职场文书
导游词之鲁迅祖居
2019/10/17 职场文书
Jackson 反序列化时实现大小写不敏感设置
2021/06/29 Java/Android