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基于Tkinter库实现简单文本编辑器实例
May 05 Python
利用python爬取散文网的文章实例教程
Jun 18 Python
Python中单例模式总结
Feb 20 Python
python爬虫获取新浪新闻教学
Dec 23 Python
Python 判断图像是否读取成功的方法
Jan 26 Python
python可视化text()函数使用详解
Feb 11 Python
新版Pycharm中Matplotlib不会弹出独立的显示窗口的问题
Jun 02 Python
详解使用Python写一个向数据库填充数据的小工具(推荐)
Sep 11 Python
python中的测试框架
Nov 13 Python
用python-webdriver实现自动填表的示例代码
Jan 13 Python
详解Python常用的魔法方法
Jun 03 Python
一小时学会TensorFlow2之基本操作2实例代码
Sep 04 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
分页显示Oracle数据库记录的类之二
2006/10/09 PHP
PHP使用递归按层级查找数据的方法
2019/11/10 PHP
javascript对象的property和prototype是这样一种关系
2007/03/24 Javascript
JavaScript Event学习第七章 事件属性
2010/02/07 Javascript
使用js 设置url参数
2013/07/08 Javascript
js 浏览本地文件夹系统示例代码
2013/10/24 Javascript
JavaScript与JQUERY获取元素的宽、高和位置
2017/02/26 Javascript
angularjs路由传值$routeParams详解
2020/09/05 Javascript
Angular4.x Event (DOM事件和自定义事件详解)
2018/10/09 Javascript
vue实现多级菜单效果
2019/10/19 Javascript
Node.js+Vue脚手架环境搭建的方法步骤
2020/03/08 Javascript
jQuery实现的上拉刷新功能组件示例
2020/05/01 jQuery
在vue中封装的弹窗组件使用队列模式实现方法
2020/07/23 Javascript
wxPython窗口的继承机制实例分析
2014/09/28 Python
python实现堆栈与队列的方法
2015/01/15 Python
Django中的“惰性翻译”方法的相关使用
2015/07/27 Python
Python 正则表达式实现计算器功能
2017/04/29 Python
Python实现Linux的find命令实例分享
2017/06/04 Python
tensorflow实现对图片的读取的示例代码
2018/02/12 Python
Python多线程原理与用法详解
2018/08/20 Python
PyQt5实现五子棋游戏(人机对弈)
2020/03/24 Python
python初学者,用python实现基本的学生管理系统(python3)代码实例
2019/04/10 Python
ansible-playbook实现自动部署KVM及安装python3的详细教程
2020/05/11 Python
解决python执行较大excel文件openpyxl慢问题
2020/05/15 Python
html5 Canvas画图教程(6)—canvas里画曲线之arcTo方法
2013/01/09 HTML / CSS
美国最大的网络男装服装品牌:Bonobos
2017/05/25 全球购物
Belvilla德国:在线预订度假屋
2018/04/10 全球购物
营销与策划专业毕业生求职信
2013/11/01 职场文书
新三好学生主要事迹
2014/01/23 职场文书
党支部书记岗位责任制
2014/02/11 职场文书
2015元旦晚会主持人开场白+结束语
2014/12/14 职场文书
工程部主管岗位职责
2015/02/12 职场文书
2019年行政人事个人工作总结范本!
2019/07/19 职场文书
Python虚拟环境virtualenv是如何使用的
2021/06/20 Python
python模拟浏览器 使用selenium进入好友QQ空间并留言
2022/04/12 Python
安装harbor作为docker镜像仓库的问题
2022/06/14 Servers