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 相关文章推荐
python实现数通设备tftp备份配置文件示例
Apr 02 Python
Python面向对象编程中关于类和方法的学习笔记
Jun 30 Python
Python使用requests及BeautifulSoup构建爬虫实例代码
Jan 24 Python
Python+OpenCV实现车牌字符分割和识别
Mar 31 Python
详解pandas的外部数据导入与常用方法
May 01 Python
一文了解Python并发编程的工程实现方法
May 31 Python
python如何获取列表中每个元素的下标位置
Jul 01 Python
详解python中的生成器、迭代器、闭包、装饰器
Aug 22 Python
python路径的写法及目录的获取方式
Dec 26 Python
Python反爬虫伪装浏览器进行爬虫
Feb 28 Python
拿来就用!Python批量合并PDF的示例代码
Aug 10 Python
python os.rename实例用法详解
Dec 06 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代码
2010/08/08 PHP
PHP正则表达式替换站点关键字链接后空白的解决方法
2014/09/16 PHP
CI框架安全类Security.php源码分析
2014/11/04 PHP
Zend Framework生成验证码并实现验证码验证功能(附demo源码下载)
2016/03/22 PHP
利用php输出不同的心形图案
2016/04/22 PHP
PHP实现的统计数据功能详解
2016/12/06 PHP
php 删除指定文件夹的实例讲解
2017/07/25 PHP
PHP使用观察者模式处理异常信息的方法详解
2019/09/24 PHP
JavaScript网页制作特殊效果用随机数
2007/05/22 Javascript
模仿JQuery.extend函数扩展自己对象的js代码
2009/12/09 Javascript
浅析document.createDocumentFragment()与js效率
2013/07/08 Javascript
jQuery中通过ajax调用webservice传递数组参数的问题实例详解
2016/05/20 Javascript
JS集成fckeditor及判断内容是否为空的方法
2016/05/27 Javascript
javascript实现日期三级联动下拉框选择菜单
2020/12/03 Javascript
webpack实现热更新(实施同步刷新)
2017/07/28 Javascript
Angular客户端请求Rest服务跨域问题的解决方法
2017/09/19 Javascript
React diff算法的实现示例
2018/04/20 Javascript
js简单的分页器插件代码实例
2019/09/11 Javascript
原生JavaScript创建不可变对象的方法简单示例
2020/05/07 Javascript
解决vue bus.$emit触发第一次$on监听不到问题
2020/07/28 Javascript
讲解Python中的递归函数
2015/04/27 Python
Python爬虫实现全国失信被执行人名单查询功能示例
2018/05/03 Python
Python简单获取网卡名称及其IP地址的方法【基于psutil模块】
2018/05/24 Python
对Python 窗体(tkinter)树状数据(Treeview)详解
2018/10/11 Python
对pandas处理json数据的方法详解
2019/02/08 Python
pyqt弹出新对话框,以及关闭对话框获取数据的实例
2019/06/18 Python
安装docker-compose的两种最简方法
2019/07/30 Python
Python调用.NET库的方法步骤
2019/12/27 Python
基于python实现把json数据转换成Excel表格
2020/05/07 Python
html5定位获取当前位置并在百度地图上显示
2014/08/22 HTML / CSS
美国知名运动产品零售商:Foot Locker
2016/07/23 全球购物
兰蔻美国官网:Lancome美国
2017/04/25 全球购物
运动会广播稿50字
2014/01/26 职场文书
法英专业大学生职业生涯规划书范文
2014/09/22 职场文书
同学聚会邀请函
2015/01/30 职场文书
CSS3实现列表无限滚动/轮播效果
2021/06/23 HTML / CSS