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实现BT种子和磁力链接的相互转换
Nov 09 Python
python实现爬虫统计学校BBS男女比例之数据处理(三)
Dec 31 Python
Python字符串和字典相关操作的实例详解
Sep 23 Python
使用pandas的DataFrame的plot方法绘制图像的实例
May 24 Python
Python中三元表达式的几种写法介绍
Mar 04 Python
详解python中@的用法
Mar 27 Python
python+OpenCV实现车牌号码识别
Nov 08 Python
如何配置关联Python 解释器 Anaconda的教程(图解)
Apr 30 Python
python 绘制正态曲线的示例
Sep 24 Python
Python运算符+与+=的方法实例
Feb 18 Python
Opencv 图片的OCR识别的实战示例
Mar 02 Python
你需要掌握的20个Python常用技巧
Feb 28 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
Email+URL的判断和自动转换函数
2006/10/09 PHP
PHP strip_tags() 去字符串中的 HTML、XML 以及 PHP 标签的函数
2016/05/22 PHP
PHP模型Model类封装数据库操作示例
2019/03/14 PHP
PHP查找一列有序数组是否包含某值的方法
2020/02/07 PHP
jQuery-ui中自动完成实现方法
2010/06/10 Javascript
Textarea与懒惰渲染实现代码
2012/01/04 Javascript
JavaScript实现点击文字切换登录窗口的方法
2015/05/11 Javascript
jQuery删除节点用法示例(remove方法)
2016/09/08 Javascript
js实现点击每个li节点,都弹出其文本值及修改
2016/12/15 Javascript
Javascript 对cookie操作详解及实例
2016/12/29 Javascript
详解vue2 $watch要注意的问题
2017/09/08 Javascript
基于Bootstrap框架菜鸟入门教程(推荐)
2017/09/17 Javascript
js实现上传并压缩图片效果
2018/01/10 Javascript
解决vue多个路由共用一个页面的问题
2018/03/12 Javascript
详解从NodeJS搭建中间层再谈前后端分离
2018/11/13 NodeJs
jQuery简单实现根据日期计算星期几的方法
2019/01/09 jQuery
微信小程序 调用微信授权窗口相关问题解决
2019/07/25 Javascript
JS实现进度条动态加载特效
2020/03/25 Javascript
Python字符串的encode与decode研究心得乱码问题解决方法
2009/03/23 Python
Python实现FTP上传文件或文件夹实例(递归)
2017/01/16 Python
python字典操作实例详解
2017/11/16 Python
Python3匿名函数用法示例
2018/07/25 Python
python中hasattr()、getattr()、setattr()函数的使用
2019/08/16 Python
python 操作mysql数据中fetchone()和fetchall()方式
2020/05/15 Python
css3实现的下拉菜单效果示例
2014/01/22 HTML / CSS
年度考核自我鉴定
2014/02/02 职场文书
关于抽烟的检讨书
2014/02/25 职场文书
《蜗牛的奖杯》教后反思
2014/04/24 职场文书
信访工作经验交流材料
2014/05/23 职场文书
2014年专项整治工作总结
2014/11/17 职场文书
2014年教务工作总结
2014/12/03 职场文书
植物园观后感
2015/06/11 职场文书
催款函范本大全
2015/06/24 职场文书
2019入党申请书格式
2019/06/25 职场文书
浅谈Python魔法方法
2021/06/28 Java/Android
win11无线投屏在哪设置? win11无线投屏功能的使用方法
2022/04/08 数码科技