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实现问号表达式(?)的方法
Nov 27 Python
python网络编程学习笔记(四):域名系统
Jun 09 Python
Python实现从百度API获取天气的方法
Mar 11 Python
Python 多核并行计算的示例代码
Nov 07 Python
python素数筛选法浅析
Mar 19 Python
Python实现统计给定列表中指定数字出现次数的方法
Apr 11 Python
Python数据抓取爬虫代理防封IP方法
Dec 23 Python
通过python连接Linux命令行代码实例
Feb 18 Python
最简单的matplotlib安装教程(小白)
Jul 28 Python
Python创建文件夹与文件的快捷方法
Dec 08 Python
python中用ggplot绘制画图实例讲解
Jan 26 Python
python关于集合的知识案例详解
May 30 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
php生成PDF格式文件并且加密
2015/06/22 PHP
PHP使用SOAP调用API操作示例
2018/12/25 PHP
PHP+Mysql分布式事务与解决方案深入理解
2021/02/27 PHP
Highslide.js是一款基于js实现的网页中图片展示插件
2020/03/30 Javascript
jquery动画2.元素坐标动画效果(创建一个图片走廊)
2012/08/24 Javascript
js写一个弹出层并锁屏效果实现代码
2012/12/07 Javascript
JS 对输入框进行限制(常用的都有)
2013/07/30 Javascript
一个简单的JS时间控件示例代码(JS时分秒时间控件)
2013/11/22 Javascript
JS实现匀速运动的代码实例
2013/11/29 Javascript
鼠标事件的screenY,pageY,clientY,layerY,offsetY属性详解
2015/03/12 Javascript
JavaScript制作windows经典扫雷小游戏
2015/03/31 Javascript
浅谈js 闭包引起的内存泄露问题
2015/06/22 Javascript
JS+CSS实现美化的下拉列表框效果
2015/08/11 Javascript
Bootstrap Chart组件使用教程
2016/04/28 Javascript
AngularJs自定义服务之实现签名和加密
2016/08/02 Javascript
vue监听对象及对象属性问题
2018/08/20 Javascript
js获取form表单中name属性的值
2019/02/27 Javascript
详解VUE Element-UI多级菜单动态渲染的组件
2019/04/25 Javascript
如何使用Jquery动态生成二级选项列表
2020/02/06 jQuery
webpack 动态批量加载文件的实现方法
2020/03/19 Javascript
原生JS利用transform实现banner的无限滚动示例代码
2020/06/15 Javascript
简单的连接MySQL与Python的Bottle框架的方法
2015/04/30 Python
Python编程实现控制cmd命令行显示颜色的方法示例
2017/08/14 Python
VSCode Python开发环境配置的详细步骤
2019/02/22 Python
Python PyInstaller安装和使用教程详解
2020/01/08 Python
django 解决扩展自带User表遇到的问题
2020/05/14 Python
python如何输出反斜杠
2020/06/18 Python
python实现学生成绩测评系统
2020/06/22 Python
pycharm + django跨域无提示的解决方法
2020/12/06 Python
奥林匹亚体育:Olympia Sports
2020/12/30 全球购物
四年大学生活的个人自我评价
2013/12/11 职场文书
大学生毕业的自我评价分享
2014/01/02 职场文书
金融管理专业毕业生求职信
2014/03/12 职场文书
入党积极分子学习党的纲领思想汇报
2014/09/13 职场文书
2015年国庆节新闻稿
2015/07/18 职场文书
靠谱的活动总结
2019/04/16 职场文书