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中使用Beautiful Soup库的超详细教程
Apr 30 Python
python利用正则表达式提取字符串
Dec 08 Python
Python内置函数—vars的具体使用方法
Dec 04 Python
python组合无重复三位数的实例
Nov 13 Python
Pycharm以root权限运行脚本的方法
Jan 19 Python
python 字符串追加实例
Jul 20 Python
python super用法及原理详解
Jan 20 Python
在python中logger setlevel没有生效的解决
Feb 21 Python
python爬虫数据保存到mongoDB的实例方法
Jul 28 Python
python高级特性简介
Aug 13 Python
Python开发简易五子棋小游戏
May 02 Python
python 使用pandas读取csv文件的方法
Dec 24 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
phpMyAdmin 链接表的附加功能尚未激活的问题
2010/08/01 PHP
全新的PDO数据库操作类php版(仅适用Mysql)
2012/07/22 PHP
php stripslashes和addslashes的区别
2014/02/03 PHP
PHP网页游戏学习之Xnova(ogame)源码解读(五)
2014/06/23 PHP
PHP实现适用于文件内容操作的分页类
2016/06/15 PHP
thinkphp分页实现效果
2016/10/13 PHP
PHP基于递归实现的约瑟夫环算法示例
2017/08/27 PHP
PHP实现绘制二叉树图形显示功能详解【包括二叉搜索树、平衡树及红黑树】
2017/11/16 PHP
Thinkphp5行为使用方法汇总
2017/12/21 PHP
php进程daemon化的正确实现方法
2018/09/06 PHP
JS鼠标事件大全 推荐收藏
2011/11/01 Javascript
js判断ie版本号的简单实现代码
2014/03/05 Javascript
JavaScript通过元素的ID和name设置样式
2014/07/08 Javascript
js实现精确到毫秒的倒计时效果
2016/08/05 Javascript
JavaScript 监控微信浏览器且自带返回按钮时间
2016/11/27 Javascript
JS实现网页抢购功能(触发,终止脚本)
2017/11/27 Javascript
jquery.onoff实现简单的开关按钮功能(推荐)
2018/05/24 jQuery
JS实现移动端触屏拖拽功能
2018/07/31 Javascript
基于layui框架响应式布局的一些使用详解
2019/09/16 Javascript
Nodejs使用archiver-zip-encrypted库加密压缩文件时报错(解决方案)
2019/11/18 NodeJs
PHP 502bad gateway原因及解决方案
2020/11/13 Javascript
python实现的系统实用log类实例
2015/06/30 Python
python Pandas 读取txt表格的实例
2018/04/29 Python
详解python分布式进程
2018/10/08 Python
python 数据生成excel导出(xlwt,wlsxwrite)代码实例
2019/08/23 Python
通过自学python能找到工作吗
2020/06/21 Python
python利用paramiko实现交换机巡检的示例
2020/09/22 Python
python 实现批量图片识别并翻译
2020/11/02 Python
NBA欧洲商店(法国):NBA Europe Store FR
2016/10/19 全球购物
Rosetta Stone官方网站:语言学习
2019/01/05 全球购物
后勤管理员岗位职责
2014/08/27 职场文书
支部书记四风问题对照检查材料
2014/10/04 职场文书
本科毕业论文答辩稿
2015/06/23 职场文书
生日宴会祝酒词
2015/08/10 职场文书
有趣的二维码:使用MyQR和qrcode来制作二维码
2021/05/10 Python
Redis Cluster 集群搭建你会吗
2021/08/04 Redis