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 json encode datetime类型
Dec 28 Python
Python的Django框架中forms表单类的使用方法详解
Jun 21 Python
教你学会使用Python正则表达式
Sep 07 Python
Python3.5 + sklearn利用SVM自动识别字母验证码方法示例
May 10 Python
使用python爬取微博数据打造一颗“心”
Jun 28 Python
django实现模板中的字符串文字和自动转义
Mar 31 Python
Python如何脚本过滤文件中的注释
May 27 Python
python 8种必备的gui库
Aug 27 Python
Elasticsearch py客户端库安装及使用方法解析
Sep 14 Python
爬虫代理的cookie如何生成运行
Sep 22 Python
Python 的 f-string 可以连接字符串与数字的原因解析
Feb 20 Python
python opencv检测直线 cv2.HoughLinesP的实现
Jun 18 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通过curl模拟登陆DZ论坛
2015/05/11 PHP
JavaScript Tips 使用DocumentFragment加快DOM渲染速度
2010/06/28 Javascript
点击隐藏页面左栏或右栏实现js代码
2013/04/01 Javascript
文字不间断滚动(上下左右)实例代码
2013/04/21 Javascript
Node.js实现的简易网页抓取功能示例
2014/12/05 Javascript
AngularJS语法详解
2015/01/23 Javascript
AngularJS 简单应用实例
2016/07/28 Javascript
AngularJs IE Compatibility 兼容老版本IE
2016/09/01 Javascript
vue实现在一个方法执行完后执行另一个方法的示例
2018/08/25 Javascript
vue组件tabbar使用方法详解
2018/11/06 Javascript
vue远程加载sfc组件思路详解
2019/12/25 Javascript
vue中keep-alive内置组件缓存的实例代码
2020/04/16 Javascript
[56:46]2018DOTA2亚洲邀请赛 3.31 小组赛 B组 VP vs Effect
2018/04/01 DOTA
简介Python设计模式中的代理模式与模板方法模式编程
2016/02/02 Python
Python单元测试简单示例
2018/07/03 Python
对python3新增的byte类型详解
2018/12/04 Python
python三引号输出方法
2019/02/27 Python
六行python代码的爱心曲线详解
2019/05/17 Python
django自带调试服务器的使用详解
2019/08/29 Python
Python中过滤字符串列表的方法
2020/12/22 Python
关于HTML5 Placeholder新标签低版本浏览器下不兼容的问题分析及解决办法
2016/01/27 HTML / CSS
美国健康和保健平台:healtop
2020/07/02 全球购物
幼儿园庆六一游园活动方案
2014/01/29 职场文书
2014年3.15团委活动总结
2014/03/16 职场文书
《雪儿》教学反思
2014/04/17 职场文书
招股说明书范本
2014/05/06 职场文书
应届本科毕业生求职信
2014/07/23 职场文书
怎样写离婚协议书
2014/09/10 职场文书
2014国庆节商场促销活动策划方案
2014/09/16 职场文书
违反工作规定检讨书范文
2014/12/14 职场文书
模范教师材料大全
2014/12/16 职场文书
优秀员工推荐材料
2014/12/20 职场文书
2015年度党风廉政建设工作情况汇报
2015/01/02 职场文书
初中语文教学研修日志
2015/11/13 职场文书
导游词之五台山
2019/10/11 职场文书
创业计划书之养殖业
2019/10/11 职场文书