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 相关文章推荐
Eclipse和PyDev搭建完美Python开发环境教程(Windows篇)
Nov 16 Python
深入浅析Python中的yield关键字
Jan 24 Python
Python实现k-means算法
Feb 23 Python
python实现简单神经网络算法
Mar 10 Python
Python将文本去空格并保存到txt文件中的实例
Jul 24 Python
对python:threading.Thread类的使用方法详解
Jan 31 Python
python中的decimal类型转换实例详解
Jun 26 Python
python获取txt文件词向量过程详解
Jul 05 Python
Python3显示当前时间、计算时间差及时间加减法示例代码
Sep 07 Python
用python解压分析jar包实例
Jan 16 Python
如何利用python 读取配置文件
Jan 06 Python
python四种出行路线规划的实现
Jun 23 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访问查询mysql数据的三种方法
2006/10/09 PHP
dedecms模板标签代码官方参考
2007/03/17 PHP
php 网页播放器用来播放在线视频的代码(自动判断并选择视频文件类型)
2010/06/03 PHP
PHP通过内置函数memory_get_usage()获取内存使用情况
2014/11/20 PHP
5款适合PHP使用的HTML编辑器推荐
2015/07/03 PHP
详解php curl带有csrf-token验证模拟提交方法
2018/04/18 PHP
php use和include区别总结
2019/10/13 PHP
js控制frameSet示例
2013/09/10 Javascript
javascript实现数字+字母验证码的简单实例
2014/02/10 Javascript
javascript实现信息增删改查的方法
2015/07/25 Javascript
jQuery实现有动画淡出效果的二级折叠菜单代码
2015/10/17 Javascript
jQuery插件ImageDrawer.js实现动态绘制图片动画(附源码下载)
2016/02/25 Javascript
深入理解node exports和module.exports区别
2016/06/01 Javascript
JavaScript必看小技巧(必看)
2016/06/07 Javascript
js计算系统当前日期是星期几的方法
2016/07/14 Javascript
js实现图片淡入淡出切换简易效果
2016/08/22 Javascript
Nodejs下用submit提交表单提示cannot post错误的解决方法
2016/11/21 NodeJs
jquery将标签元素的高设为屏幕的百分比
2017/04/19 jQuery
jQuery EasyUI tree增加搜索功能的实现方法
2017/04/27 jQuery
浅谈对于react-thunk中间件的简单理解
2019/05/01 Javascript
vue输入节流,避免实时请求接口的实例代码
2019/10/30 Javascript
NodeJS实现一个聊天室功能
2019/11/25 NodeJs
JavaScript实现模态对话框实例
2020/01/13 Javascript
微信小程序scroll-view实现滚动到锚点左侧导航栏点餐功能(点击种类,滚动到锚点)
2020/06/11 Javascript
[01:06]DOTA2隆重推出2016冬季勇士令状 内含上海特级锦标赛互动指南
2016/02/17 DOTA
python 2.7 检测一个网页是否能正常访问的方法
2018/12/26 Python
python 监测内存和cpu的使用率实例
2019/11/28 Python
美国知名的旅游网站:OneTravel
2018/10/09 全球购物
俄罗斯珠宝市场的领导者之一:Бронницкий ювелир
2019/10/02 全球购物
教育技术职业规划范文
2014/03/04 职场文书
火箭队口号
2014/06/18 职场文书
学校综治宣传月活动总结
2014/07/02 职场文书
委托书格式
2014/08/01 职场文书
幼儿园大班开学寄语
2014/08/02 职场文书
2016年中秋祝酒词
2015/11/26 职场文书
go select编译期的优化处理逻辑使用场景分析
2021/06/28 Golang