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查找第k小元素代码分享
Dec 18 Python
Python 不同对象比较大小示例探讨
Aug 21 Python
python多线程threading.Lock锁用法实例
Nov 01 Python
python清除字符串里非字母字符的方法
Jul 02 Python
python实现飞机大战
Sep 11 Python
学生信息管理系统Python面向对象版
Jan 30 Python
python创建属于自己的单词词库 便于背单词
Jul 30 Python
关于django 1.10 CSRF验证失败的解决方法
Aug 31 Python
appium+python adb常用命令分享
Mar 06 Python
Python urlencode和unquote函数使用实例解析
Mar 31 Python
Jupyter notebook如何修改平台字体
May 13 Python
Python  Asyncio模块实现的生产消费者模型的方法
Mar 01 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
弄了个检测传输的参数是否为数字的Function
2006/12/06 PHP
mysql5详细安装教程
2007/01/15 PHP
PHP中md5()函数的用法讲解
2019/03/30 PHP
关于laravel后台模板laravel-admin select框的使用详解
2019/10/03 PHP
解密效果
2006/06/23 Javascript
用jscript启动sqlserver
2007/06/21 Javascript
JavaScript高级程序设计 阅读笔记(十八) js跨平台的事件
2012/08/14 Javascript
javascript中处理时间戳为日期格式的方法
2014/01/02 Javascript
jQuery中DOM树操作之复制元素的方法
2015/01/23 Javascript
nodejs中使用多线程编程的方法实例
2015/03/24 NodeJs
使用Raygun对Node.js应用进行错误处理的方法
2015/06/23 Javascript
bootstrap-treeview自定义双击事件实现方法
2016/01/09 Javascript
jQuery中使用animate自定义动画的方法
2016/05/29 Javascript
jQuery插件HighCharts绘制2D圆环图效果示例【附demo源码下载】
2017/03/09 Javascript
jquery实现企业定位式导航效果
2018/01/01 jQuery
js逆向解密之网络爬虫
2019/05/30 Javascript
微信小程序页面上下滚动效果
2020/11/18 Javascript
在vue中高德地图引入和轨迹的绘制的实现
2019/10/11 Javascript
javascript实现搜索筛选功能实例代码
2020/11/12 Javascript
[02:57]DOTA2亚洲邀请赛小组赛第四日 赛事回顾
2015/02/02 DOTA
python通过ElementTree操作XML获取结点读取属性美化XML
2013/12/02 Python
详解python 字符串和日期之间转换 StringAndDate
2017/05/04 Python
Python操作使用MySQL数据库的实例代码
2017/05/25 Python
Python验证文件是否可读写代码分享
2017/12/11 Python
Python Selenium Cookie 绕过验证码实现登录示例代码
2018/04/10 Python
Python3爬虫之urllib携带cookie爬取网页的方法
2018/12/28 Python
python实现转盘效果 python实现轮盘抽奖游戏
2019/01/22 Python
Python实现定时自动关闭的tkinter窗口方法
2019/02/16 Python
python dlib人脸识别代码实例
2019/04/04 Python
python批量下载抖音视频
2019/06/17 Python
python获取点击的坐标画图形的方法
2019/07/09 Python
工作失职检讨书范文
2014/01/16 职场文书
社区服务标语
2014/07/01 职场文书
2014年国庆节庆祝建国65周年比赛演讲稿
2014/09/21 职场文书
2014年仓库管理员工作总结
2014/11/18 职场文书
团队拓展训练心得体会
2016/01/12 职场文书