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使用Flask框架同时上传多个文件的方法
Mar 21 Python
Python及Django框架生成二维码的方法分析
Jan 31 Python
使用Python读取安卓手机的屏幕分辨率方法
Mar 31 Python
点球小游戏python脚本
May 22 Python
python递归函数绘制分形树的方法
Jun 22 Python
解决pandas展示数据输出时列名不能对齐的问题
Nov 18 Python
pytorch:实现简单的GAN示例(MNIST数据集)
Jan 10 Python
Scrapy框架基本命令与settings.py设置
Feb 06 Python
Java如何基于wsimport调用wcf接口
Jun 17 Python
如何使用PyCharm引入需要使用的包的方法
Sep 22 Python
python 牛顿法实现逻辑回归(Logistic Regression)
Oct 15 Python
浅析Python实现DFA算法
Jun 26 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 fsockopen写的HTTP下载的类
2007/02/22 PHP
PHP代码审核的详细介绍
2013/06/13 PHP
PHP生成json和xml类型接口数据格式
2015/05/17 PHP
php获取flash尺寸详细数据的方法
2016/11/12 PHP
PHP简单获取随机数的常用方法小结
2017/06/07 PHP
PHP基于IMAP收取邮件的方法示例
2017/08/07 PHP
laravel自定义分页的实现案例offset()和limit()
2019/10/15 PHP
基于jquery的自定义鼠标提示效果 jquery.toolTip
2010/11/14 Javascript
JavaScript面向对象(极简主义法minimalist approach)
2012/07/17 Javascript
JS特权方法定义作用以及与公有方法的区别
2013/03/18 Javascript
js添加select下默认的option的value和text的方法
2014/10/19 Javascript
JavaScript实现相册弹窗功能(zepto.js)
2016/06/21 Javascript
利用vue-router实现二级菜单内容转换
2016/11/30 Javascript
js实现百度地图定位于地址逆解析,显示自己当前的地理位置
2016/12/08 Javascript
JavaScript使用原型和原型链实现对象继承的方法详解
2017/04/05 Javascript
解决bootstrap中使用modal加载kindeditor时弹出层文本框不能输入的问题
2017/06/05 Javascript
详解从Vue.js源码看异步更新DOM策略及nextTick
2017/10/11 Javascript
vue.js 实现图片本地预览 裁剪 压缩 上传功能
2018/03/01 Javascript
微信小程序学习笔记之获取位置信息操作图文详解
2019/03/29 Javascript
JS事件流与事件处理程序实例分析
2019/08/16 Javascript
Vue.js如何使用Socket.IO的示例代码
2019/09/05 Javascript
JS页面获取 session 值,作用域和闭包学习笔记
2019/10/16 Javascript
[02:56]DOTA2矮人直升机 英雄基础教程
2013/11/26 DOTA
用python实现简单EXCEL数据统计的实例
2017/01/24 Python
机器学习python实战之手写数字识别
2017/11/01 Python
Python开发虚拟环境使用virtualenvwrapper的搭建步骤教程图解
2018/09/19 Python
django做form表单的数据验证过程详解
2019/07/26 Python
python程序 创建多线程过程详解
2019/09/23 Python
Python configparser模块配置文件过程解析
2020/03/03 Python
python 读取yaml文件的两种方法(在unittest中使用)
2020/12/01 Python
adidas马来西亚官网:adidas MY
2020/09/12 全球购物
什么是继承
2013/12/07 面试题
毕业生自我鉴定
2013/12/04 职场文书
Spring boot实现上传文件到本地服务器
2022/08/14 Java/Android
Win10本地连接不见了怎么恢复? win10系统电脑本地连接不见了解决方法
2023/01/09 数码科技