TensorFlow实现保存训练模型为pd文件并恢复


Posted in Python onFebruary 06, 2020

TensorFlow保存模型代码

import tensorflow as tf
from tensorflow.python.framework import graph_util
var1 = tf.Variable(1.0, dtype=tf.float32, name='v1')
var2 = tf.Variable(2.0, dtype=tf.float32, name='v2')
var3 = tf.Variable(2.0, dtype=tf.float32, name='v3')
x = tf.placeholder(dtype=tf.float32, shape=None, name='x')
x2 = tf.placeholder(dtype=tf.float32, shape=None, name='x2')
addop = tf.add(x, x2, name='add')
addop2 = tf.add(var1, var2, name='add2')
addop3 = tf.add(var3, var2, name='add3')
initop = tf.global_variables_initializer()
model_path = './Test/model.pb'
with tf.Session() as sess:
  sess.run(initop)
  print(sess.run(addop, feed_dict={x: 12, x2: 23}))
  output_graph_def = graph_util.convert_variables_to_constants(sess, sess.graph_def, ['add', 'add2', 'add3'])
  # 将计算图写入到模型文件中
  model_f = tf.gfile.FastGFile(model_path, mode="wb")
  model_f.write(output_graph_def.SerializeToString())

读取模型代码

import tensorflow as tf
with tf.Session() as sess:
  model_f = tf.gfile.FastGFile("./Test/model.pb", mode='rb')
  graph_def = tf.GraphDef()
  graph_def.ParseFromString(model_f.read())
  c = tf.import_graph_def(graph_def, return_elements=["add2:0"])
  c2 = tf.import_graph_def(graph_def, return_elements=["add3:0"])
  x, x2, c3 = tf.import_graph_def(graph_def, return_elements=["x:0", "x2:0", "add:0"])

  print(sess.run(c))
  print(sess.run(c2))
  print(sess.run(c3, feed_dict={x: 23, x2: 2}))

以上这篇TensorFlow实现保存训练模型为pd文件并恢复就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
简单介绍Python中的filter和lambda函数的使用
Apr 07 Python
Python搭建FTP服务器的方法示例
Jan 19 Python
浅谈使用Python内置函数getattr实现分发模式
Jan 22 Python
python数字图像处理实现直方图与均衡化
May 04 Python
python 使用正则表达式按照多个空格分割字符的实例
Dec 20 Python
python itchat给指定联系人发消息的方法
Jun 11 Python
Python PIL读取的图像发生自动旋转的实现方法
Jul 05 Python
python加密解密库cryptography使用openSSL生成的密匙加密解密
Feb 11 Python
python3 xpath和requests应用详解
Mar 06 Python
python异常处理之try finally不报错的原因
May 18 Python
浅析python 通⽤爬⾍和聚焦爬⾍
Sep 28 Python
Python基于百度AI实现抓取表情包
Jun 27 Python
解决TensorFlow模型恢复报错的问题
Feb 06 #Python
TensorFlow加载模型时出错的解决方式
Feb 06 #Python
基于Python实现人脸自动戴口罩系统
Feb 06 #Python
python时间日期操作方法实例小结
Feb 06 #Python
python通过matplotlib生成复合饼图
Feb 06 #Python
tensorflow 初始化未初始化的变量实例
Feb 06 #Python
python中count函数简单的实例讲解
Feb 06 #Python
You might like
动漫女神老婆无限好,但日本女生可能就不是这么一回事了!
2020/03/04 日漫
Thinkphp搜索时首页分页和搜索页保持条件分页的方法
2014/12/05 PHP
简单谈谈PHP中的include、include_once、require以及require_once语句
2016/04/23 PHP
基于PHP实现数据分页显示功能
2016/05/26 PHP
laravel excel 上传文件保存到本地服务器功能
2019/11/14 PHP
PHP时间相关常用函数用法示例
2020/06/03 PHP
javascript 在网页中的运用(asp.net)
2009/11/23 Javascript
用js实现trim()的解决办法
2013/04/16 Javascript
jquery动态增加text元素以及删除文本内容实例代码
2013/07/01 Javascript
js 编码转换 gb2312 和 utf8 互转的2种方法
2013/08/07 Javascript
Javascript表格翻页效果的具体实现
2013/10/05 Javascript
JS检测输入字符是否包含非法字符的示例代码
2014/02/11 Javascript
asp.net+js实现金额格式化
2015/02/27 Javascript
基于jquery css3实现点击动画弹出表单源码特效
2015/08/31 Javascript
个人网站留言页面(前端jQuery编写、后台php读写MySQL)
2016/05/03 Javascript
Bootstrap select下拉联动(jQuery cxselect)
2017/01/04 Javascript
JavaScript中offsetWidth的bug及解决方法
2017/05/17 Javascript
基于react组件之间的参数传递(详解)
2017/09/05 Javascript
angular4模块中给标签添加背景图的实现方法
2017/09/15 Javascript
python绘制简单折线图代码示例
2017/12/19 Python
Python及Django框架生成二维码的方法分析
2018/01/31 Python
python 读取文件并把矩阵转成numpy的两种方法
2019/02/12 Python
python正则表达式匹配IP代码实例
2019/12/28 Python
CSS3圆角边框和边界图片效果实例
2016/07/01 HTML / CSS
Rosetta Stone官方网站:语言学习
2019/01/05 全球购物
什么是网络协议
2016/04/07 面试题
《乡愁》教学反思
2014/02/18 职场文书
小学数学教研活动总结
2014/07/01 职场文书
安全保卫工作竞聘材料
2014/08/25 职场文书
教师查摆问题及整改措施
2014/10/11 职场文书
财务会计岗位职责
2015/02/03 职场文书
试用期辞职信范文
2015/03/02 职场文书
2015年八一建军节活动总结
2015/03/20 职场文书
入党转正介绍人意见
2015/06/03 职场文书
如何理解Vue前后端数据交互与显示
2021/05/10 Vue.js
JS实现数组去重的11种方法总结
2022/04/04 Javascript