TensorFlow实现checkpoint文件转换为pb文件


Posted in Python onFebruary 10, 2020

由于项目需要,需要将TensorFlow保存的模型从ckpt文件转换为pb文件。

import os
from tensorflow.python import pywrap_tensorflow
from net2use import inception_resnet_v2_small#这里使用自己定义的模型函数即可
import tensorflow as tf
if __name__=='__main__':
  pb_file = "./model/output.pb"
  ckpt_file = "./model/model.ckpt-652900"
  '''
这里的节点名字可能跟设想的有出入,最直接的方法是直接输出ckpt中保存的节点名字,然后对应着找节点名字,具体的进入convert_variables_to_constants函数的实现中graph_util_impl.py,130行的函数:_assert_nodes_are_present 添加代码
  print('在图中的节点是:')
  for din in name_to_node:
    print('{},在图中'.format(din))
然后运行代码,若正确就会直接保存;若失败则会保存失败,找好输出节点的名字,在output_node_names 中添加就好
'''
  output_node_names = ["embedding"]

  with tf.name_scope('input'):
    image = tf.placeholder(tf.float32,shape=(None,79,199,1),name='input_image')


  net, endpoints=inception_resnet_v2_small(image, is_training=False)
  embedding = tf.nn.l2_normalize(net,1,1e-10,name='embedding')

  config=tf.ConfigProto(allow_soft_placement=True)
  config.gpu_options.per_process_gpu_memory_fraction = 0.45
  sess = tf.Session(config = config)
  saver = tf.train.Saver()
  saver.restore(sess, ckpt_file)
  print('read success')
  converted_graph_def = tf.graph_util.convert_variables_to_constants(sess,
                input_graph_def = sess.graph.as_graph_def(),
                output_node_names = output_node_names)

  with tf.gfile.GFile(pb_file, "wb") as f:
    f.write(converted_graph_def.SerializeToString())

  print('保存成功')

以上这篇TensorFlow实现checkpoint文件转换为pb文件就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
pycharm 使用心得(七)一些实用功能介绍
Jun 06 Python
零基础写python爬虫之爬虫的定义及URL构成
Nov 04 Python
python从sqlite读取并显示数据的方法
May 08 Python
python实现简单点对点(p2p)聊天
Sep 13 Python
Python内置函数—vars的具体使用方法
Dec 04 Python
Python实现读取Properties配置文件的方法
Mar 29 Python
Python的numpy库中将矩阵转换为列表等函数的方法
Apr 04 Python
Python 安装第三方库 pip install 安装慢安装不上的解决办法
Jun 18 Python
Python基于WordCloud制作词云图
Nov 29 Python
Pytorch 数据加载与数据预处理方式
Dec 31 Python
python 实现在shell窗口中编写print不向屏幕输出
Feb 19 Python
Python第三方包PrettyTable安装及用法解析
Jul 08 Python
Django关于admin的使用技巧和知识点
Feb 10 #Python
Python实现括号匹配方法详解
Feb 10 #Python
Python re正则表达式元字符分组()用法分享
Feb 10 #Python
Tensorflow: 从checkpoint文件中读取tensor方式
Feb 10 #Python
Python获取、格式化当前时间日期的方法
Feb 10 #Python
Python3操作MongoDB增册改查等方法详解
Feb 10 #Python
Python实现结构体代码实例
Feb 10 #Python
You might like
一个简单至极的PHP缓存类代码
2015/10/23 PHP
php 解析xml 的四种方法详细介绍
2016/10/26 PHP
PHP正则匹配反斜杠'\'和美元'$'的方法
2017/02/08 PHP
php删除数组指定元素实现代码
2017/05/03 PHP
微信公众平台开发教程①获取用户Openid及个人信息图文详解
2019/04/10 PHP
jQuery toggle()设置CSS样式
2009/11/05 Javascript
快速排序 php与javascript的不同之处
2011/02/22 Javascript
推荐10个2014年最佳的jQuery视频插件
2014/11/12 Javascript
jQuery使用load()方法载入另外一个网页文件内的指定标签内容到div标签的方法
2015/03/25 Javascript
详解javascript数组去重问题
2015/11/06 Javascript
js实现搜索框关键字智能匹配代码
2020/03/26 Javascript
js实现五星评价功能
2017/03/08 Javascript
深入浅析ES6 Class 中的 super 关键字
2017/10/20 Javascript
JavaScript实现获取两个排序数组的中位数算法示例
2019/02/26 Javascript
如何在微信小程序中存setStorage
2019/12/13 Javascript
Vue实现跑马灯效果
2020/05/25 Javascript
详解Vue数据驱动原理
2020/11/17 Javascript
python调用java的Webservice示例
2014/03/10 Python
python使用正则表达式匹配字符串开头并打印示例
2017/01/11 Python
Python新手入门最容易犯的错误总结
2017/04/24 Python
Python比较2个时间大小的实现方法
2018/04/10 Python
Python实现抓取HTML网页并以PDF文件形式保存的方法
2018/05/08 Python
一个可以套路别人的python小程序实例代码
2019/04/09 Python
pyqt5 comboBox获得下标、文本和事件选中函数的方法
2019/06/14 Python
python 计算两个列表的相关系数的实现
2019/08/29 Python
PyCharm第一次安装及使用教程
2020/01/08 Python
Python Opencv 通过轨迹(跟踪)栏实现更改整张图像的背景颜色
2020/03/09 Python
比利时家具购买网站:Home24
2019/01/03 全球购物
中科前程Java笔试题
2016/11/20 面试题
教师找工作推荐信
2013/11/23 职场文书
大学生求职信范文应怎么写
2014/01/01 职场文书
致200米运动员广播稿
2014/02/06 职场文书
怎样写好创业计划书的内容
2014/02/06 职场文书
2014庆六一活动方案
2014/03/02 职场文书
爱国主义教育活动总结
2014/05/07 职场文书
利用python进行数据加载
2021/06/20 Python