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采用raw_input读取输入值的方法
Aug 18 Python
python中readline判断文件读取结束的方法
Nov 08 Python
python base64 decode incorrect padding错误解决方法
Jan 08 Python
基于python的字节编译详解
Sep 20 Python
python二维列表一维列表的互相转换实例
Jul 02 Python
python每天定时运行某程序代码
Aug 16 Python
Python 实现训练集、测试集随机划分
Jan 08 Python
Python定义函数实现累计求和操作
May 03 Python
PyCharm中如何直接使用Anaconda已安装的库
May 28 Python
django ObjectDoesNotExist 和 DoesNotExist的用法
Jul 09 Python
Python使用jpype模块调用jar包过程解析
Jul 29 Python
Python 读取千万级数据自动写入 MySQL 数据库
Jun 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
十天学会php之第六天
2006/10/09 PHP
php array_search() 函数使用
2010/04/13 PHP
php中 ob_start等函数截取标准输出的方法
2015/06/22 PHP
JS+PHP实现用户输入数字后显示最大的值及所在位置
2017/06/19 PHP
PHP通过GD库实现验证码功能示例
2019/02/23 PHP
在Laravel5中正确设置文件权限的方法
2019/05/22 PHP
获取页面高度,窗口高度,滚动条高度等参数值getPageSize,getPageScroll
2006/09/22 Javascript
Javascript 遍历对象中的子对象
2009/07/03 Javascript
IE与Firefox在JavaScript上的7个不同写法小结
2009/09/14 Javascript
js 操作符实例代码
2009/10/24 Javascript
jQuery插件开发全解析
2012/10/10 Javascript
iframe异步加载实现点击左边菜单加载右边内容实例讲解
2013/03/04 Javascript
关于js注册事件的常用方法
2013/04/03 Javascript
JavaScript实现查找字符串中第一个不重复的字符
2014/12/29 Javascript
js简单实现点击左右运动的方法
2015/04/10 Javascript
深入探究使JavaScript动画流畅的一些方法
2015/06/30 Javascript
jQuery Easyui 验证两次密码输入是否相等
2016/05/13 Javascript
JS判断字符串变量是否含有某个字串的实现方法
2016/06/03 Javascript
基于Vue渲染与插件的加载顺序的问题详解
2018/03/05 Javascript
JQuery元素快速查找与操作
2018/04/22 jQuery
对vue中v-on绑定自定事件的实例讲解
2018/09/06 Javascript
H5+C3+JS实现五子棋游戏(AI篇)
2020/05/28 Javascript
JavaScript indexOf()原理及使用方法详解
2020/07/09 Javascript
Python Tkinter 简单登录界面的实现
2019/06/14 Python
anaconda如何查看并管理python环境
2019/07/05 Python
PyQt5中多线程模块QThread使用方法的实现
2020/01/31 Python
python与c语言的语法有哪些不一样的
2020/09/13 Python
python批量提取图片信息并保存的实现
2021/02/05 Python
纯CSS3实现的井字棋游戏
2020/11/25 HTML / CSS
车间班长岗位职责
2013/11/30 职场文书
小班上学期评语
2014/05/05 职场文书
员工试用期自我评价
2014/09/18 职场文书
2014年领导班子专项整治整改方案
2014/09/28 职场文书
2014年驻村干部工作总结
2014/11/17 职场文书
2014工程部年度工作总结
2014/12/17 职场文书
实习证明模板
2015/06/16 职场文书