tensorflow没有output结点,存储成pb文件的例子


Posted in Python onJanuary 04, 2020

Tensorflow中保存成pb file 需要 使用函数

graph_util.convert_variables_to_constants(sess, sess.graph_def,

output_node_names=[]) []中需要填写你需要保存的结点。如果保存的结点在神经网络中没有被显示定义该怎么办?

例如我使用了tf.contrib.slim或者keras,在tf的高层很多情况下都会这样。

在写神经网络时,只需要简单的一层层传导,一个slim.conv2d层就包含了kernal,bias,activation function,非常的方便,好处是网络结构一目了然,坏处是什么呢?

tensorflow没有output结点,存储成pb文件的例子

在尝试保存pb的 output node names时,需要将最后的输出结点保存下来,与这个结点相关的,从输入开始,经过层层传递的嵌套函数或者操作的相关结点,都会被保存,但无效的例如 计算准确率,计算loss等,就可以省略了,因为保存的pb主要是用来做预测的。

在准备查看所有的结点名称并选取保存时,发现scope "local3"里面仅有相关的weights 和biases,这两个是单独存在的,即保存这两个参数并没有任何意义。

tensorflow没有output结点,存储成pb文件的例子

那么这时候有两种解决办法:

方法一:

graph_util.convert_variables_to_constants(sess, sess.graph_def, output_node_names=[var.name[:-2] for var in tf.global_variables()])

那么这个的意思是所有的variable的都被保存下来 但函数中要求的是 node name 我们通过 global_variables获得的是 变量名 并不是 节点名

(例如 output:0 就是变量名,又叫tensor name)

output就是 node name了。

在tensorboard中可以一窥究竟

tensorflow没有output结点,存储成pb文件的例子

通过这样 也可以将 所有的变量全部保存下来(但是你并不能使用,是因为你的output并没有名字,所以你不可以通过常用的sess.graph.get_tensor_by_name来使用)

方法二:

那就是直接改写神经网络了....当然了还是比较简单的,只要改写最后一个,改写成output即可,tensorflow中无论是 变量、操作op、函数、都可以命名,那么这个地方是一个简单的全连接,仅需要将weights*net(上一层的输出) +bias 即可,我们只要将bias相加的结果命名为 ouput即可:

with tf.name_scope('local3'):
 
  local3_weights = tf.Variable(tf.truncated_normal([4096, self.output_size], stddev=0.1))
 
  local3_bias = tf.Variable(tf.constant(0.1, shape=[self.output_size]))
 
result = tf.add(tf.matmul(net, local3_weights), local3_bias, name="output")

这样将上述的convert_variables_to_constants中的output_node_names只需要填写一个['output']即可,因为这一个output结点,需要从input开始,将所有的神经网络前向传播的操作和参数全部保存下来,因此保存的结点数量 和 方法一保存的结点数量是一样的(console显示都是 convert 24)。

完整的pb保存为:(我是将ckpt读入进来,然后存成pb的)

from tensorflow.python.platform import gfile
 
 
 
load_ckpt():
 
  path = './data/output/loss1.0/'
 
  print("read from ckpt")
 
  ckpt = tf.train.get_checkpoint_state(path)
 
  saver = tf.train.Saver()
 
  saver.restore(sess, ckpt.model_checkpoint_path)
 
 
 
def write2pb_file():
 
  constant_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def,
 
    output_node_names=["output"])
 
  with tf.gfile.GFile(path+'loss1.0.pb', mode='wb') as f:
 
  f.write(constant_graph.SerializeToString())
 
  print("Model is saved as " + path+'loss1.0.pb')
 
 
 
def main():
 
  load_ckpt()
 
  write2pb_file()

如果是简单的直接保存,那就更简单了。

pb文件的read,很多人会将一个net写成一个类,在引入的时候会将新建这个类,然后读入ckpt文件,这完全没有问题,但是在读取pb时,就会发生问题,因为pb中已经包含了图与参数,引入时会创建一个默认的图,但是net类中自己也会创建一个图,那么这时候你运行程序,参数其实并没有使用.pb的文件。

所以我们不能创建net类,然后直接读入.pb文件,对.pb文件,通过如下代码,获取.pb的graph中的输入和输出。

self.output = self.sess.graph.get_tensor_by_name("output:0")
 
self.input = self.sess.graph.get_tensor_by_name("images:0")

注意此时要加:0 因为你获取的不再是结点了,而是一个真实的变量,我的理解是,结点相当于一个类,:0是对象,默认初始化值就是对象的初始化。

然后就可以通过self.sess.run(self.output(feed_dict={self.input: your_input})))运行你的网络了!

以上这篇tensorflow没有output结点,存储成pb文件的例子就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python 正则表达式 概述及常用字符
May 04 Python
Python 列表(List)操作方法详解
Mar 11 Python
在Python的Flask框架下收发电子邮件的教程
Apr 21 Python
python装饰器初探(推荐)
Jul 21 Python
Python中str is not callable问题详解及解决办法
Feb 10 Python
Python基于多线程实现抓取数据存入数据库的方法
Jun 22 Python
pyqt弹出新对话框,以及关闭对话框获取数据的实例
Jun 18 Python
使用Python来做一个屏幕录制工具的操作代码
Jan 18 Python
解决python3.x安装numpy成功但import出错的问题
Nov 17 Python
python3代码输出嵌套式对象实例详解
Dec 03 Python
Pycharm 解决自动格式化冲突的设置操作
Jan 15 Python
matplotlib之pyplot模块之标题(title()和suptitle())
Feb 22 Python
TensorFlow查看输入节点和输出节点名称方式
Jan 04 #Python
根据tensor的名字获取变量的值方式
Jan 04 #Python
将tensorflow.Variable中的某些元素取出组成一个新的矩阵示例
Jan 04 #Python
tensorflow实现tensor中满足某一条件的数值取出组成新的tensor
Jan 04 #Python
对tensorflow中的strides参数使用详解
Jan 04 #Python
tensorflow之获取tensor的shape作为max_pool的ksize实例
Jan 04 #Python
TensorFlow tf.nn.max_pool实现池化操作方式
Jan 04 #Python
You might like
PHP树的代码,可以嵌套任意层
2006/10/09 PHP
Session保存到数据库的php类分享
2011/10/24 PHP
PHP判断一个数组是另一个数组子集的方法详解
2017/07/31 PHP
Laravel 连接(Join)示例
2019/10/16 PHP
有关javascript的性能优化 (repaint和reflow)
2013/04/12 Javascript
YUI模块开发原理详解
2013/11/18 Javascript
ExtJS判断IE浏览器类型的方法
2014/02/10 Javascript
js+HTML5基于过滤器从摄像头中捕获视频的方法
2015/06/16 Javascript
jQuery自动完成插件completer附源码下载
2016/01/04 Javascript
AngularJS向后端ASP.NET API控制器上传文件
2016/02/03 Javascript
jQuery添加options点击事件并传值实例代码
2016/05/18 Javascript
javascript 删除数组元素和清空数组的简单方法
2017/02/24 Javascript
Javascript中click与blur事件的顺序详析
2017/04/25 Javascript
Vue2.0父子组件传递函数的教程详解
2017/10/16 Javascript
Spring boot 和Vue开发中CORS跨域问题解决
2018/09/05 Javascript
vue elementUI tree树形控件获取父节点ID的实例
2018/09/12 Javascript
vue如何使用async、await实现同步请求
2019/12/09 Javascript
js实现登录拖拽窗口
2020/02/10 Javascript
python开发之函数定义实例分析
2015/11/12 Python
Python正确重载运算符的方法示例详解
2017/08/27 Python
Python+matplotlib实现华丽的文本框演示代码
2018/01/22 Python
使用python绘制cdf的多种实现方法
2020/02/25 Python
基于Python实现视频的人脸融合功能
2020/06/12 Python
SIDESTEP荷兰:在线购买鞋子
2019/11/18 全球购物
客服主管岗位职责
2013/12/13 职场文书
车间班组长的职责
2013/12/13 职场文书
教师旷工检讨书
2014/01/18 职场文书
美国探亲签证邀请信
2014/02/05 职场文书
致裁判员加油稿
2014/02/08 职场文书
离婚协议书范本及离婚须知
2014/10/15 职场文书
学校运动会感想
2015/08/10 职场文书
教学副校长工作总结
2015/08/13 职场文书
2016教师给学生的毕业寄语
2015/12/04 职场文书
发工资啦!教你用Python实现邮箱自动群发工资条
2021/05/10 Python
浅谈Python中的函数(def)及参数传递操作
2021/05/25 Python
阿里云服务器部署RabbitMQ集群的详细教程
2022/06/01 Servers