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 快速排序代码
Nov 23 Python
Python 执行字符串表达式函数(eval exec execfile)
Aug 11 Python
Python卸载模块的方法汇总
Jun 07 Python
python爬虫入门教程--快速理解HTTP协议(一)
May 25 Python
Python实现读取TXT文件数据并存进内置数据库SQLite3的方法
Aug 08 Python
Python字符串格式化的方法(两种)
Sep 19 Python
Python实现模拟登录网易邮箱的方法示例
Jul 05 Python
Win10下python 2.7.13 安装配置方法图文教程
Sep 18 Python
[原创]Python入门教程5. 字典基本操作【定义、运算、常用函数】
Nov 01 Python
python+numpy按行求一个二维数组的最大值方法
Jul 09 Python
python 叠加等边三角形的绘制的实现
Aug 14 Python
Python 转换文本编码实现解析
Aug 27 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 fsockopen中多线程问题的解决办法[翻译]
2011/11/09 PHP
关于扩展 Laravel 默认 Session 中间件导致的 Session 写入失效问题分析
2016/01/08 PHP
Laravel框架实现的上传图片到七牛功能详解
2019/09/06 PHP
js实现飞入星星特效代码
2014/10/17 Javascript
jQuery中parentsUntil()方法用法实例
2015/01/07 Javascript
javascript编写贪吃蛇游戏
2015/07/07 Javascript
浅谈JavaScript中null和undefined
2015/07/09 Javascript
情人节单身的我是如何在敲完代码之后收到12束玫瑰的(javascript)
2015/08/21 Javascript
JS+CSS实现的蓝色table选项卡效果
2015/10/08 Javascript
AngularJS使用指令增强标准表单元素功能
2016/07/01 Javascript
AngularJS实现给动态生成的元素绑定事件的方法
2016/12/14 Javascript
微信小程序 滚动到某个位置添加class效果实现代码
2017/04/19 Javascript
bootstrap table单元格新增行并编辑
2017/05/19 Javascript
微信小程序form表单组件示例代码
2018/07/15 Javascript
对layui中table组件工具栏的使用详解
2019/09/19 Javascript
小程序实现可拖动的悬浮按钮
2020/09/07 Javascript
python检测lvs real server状态
2014/01/22 Python
python中stdout输出不缓存的设置方法
2014/05/29 Python
Python生成短uuid的方法实例详解
2018/05/29 Python
python 3.6.4 安装配置方法图文教程
2018/09/18 Python
详解从Django Rest Framework响应中删除空字段
2019/01/11 Python
Opencv实现抠图背景图替换功能
2019/05/21 Python
使用Python函数进行模块化的实现
2019/11/15 Python
Chemist Warehouse官方海外旗舰店:澳洲第一连锁大药房
2017/08/25 全球购物
德国净水壶和滤芯品牌:波尔德PearlCo(家用净水器)
2020/04/29 全球购物
几个Shell Script面试题
2012/08/31 面试题
小学毕业家长寄语
2014/01/19 职场文书
大学校务公开实施方案
2014/03/31 职场文书
作文评语大全
2014/04/23 职场文书
土地转让协议书
2014/09/27 职场文书
2014年党员干部四风问题自我剖析材料
2014/09/29 职场文书
2014年文秘工作总结
2014/11/25 职场文书
工作岗位职责范本
2015/02/15 职场文书
2015年销售助理工作总结
2015/05/11 职场文书
C3 线性化算法与 MRO之Python中的多继承
2021/10/05 Python
python编程学习使用管道Pipe编写优化代码
2021/11/20 Python