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中的__getitem__方法与slice对象的切片操作
Jun 27 Python
Zabbix实现微信报警功能
Oct 09 Python
python操作mysql数据库
Mar 05 Python
Python使用wget实现下载网络文件功能示例
May 31 Python
python生成ppt的方法
Jun 07 Python
在python中使用requests 模拟浏览器发送请求数据的方法
Dec 26 Python
python 利用文件锁单例执行脚本的方法
Feb 19 Python
Opencv实现抠图背景图替换功能
May 21 Python
PyTorch的深度学习入门教程之构建神经网络
Jun 27 Python
django获取from表单multiple-select的value和id的方法
Jul 19 Python
Python爬虫实现模拟点击动态页面
Mar 05 Python
python基于机器学习预测股票交易信号
May 25 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
虫族 Zerg 历史背景
2020/03/14 星际争霸
一些常用的php函数
2006/12/06 PHP
php 伪造本地文件包含漏洞的代码
2011/11/03 PHP
PHP英文字母大小写转换函数小结
2014/05/03 PHP
PHP使用CURL获取302跳转后的地址实例
2014/05/04 PHP
PHP实现多维数组转字符串和多维数组转一维数组的方法
2015/08/08 PHP
php 5.4 全新的代码复用Trait详解
2017/01/05 PHP
php基于环形链表解决约瑟夫环问题示例
2017/11/07 PHP
laravel框架数据库操作、查询构建器、Eloquent ORM操作实例分析
2019/12/20 PHP
如何用js控制css中的float的代码
2007/08/16 Javascript
一个级联菜单代码学习及removeClass与addClass的应用
2013/01/24 Javascript
js计算文本框输入的字符数
2015/10/23 Javascript
js实现界面向原生界面发消息并跳转功能
2016/11/22 Javascript
vue-cli单页应用改成多页应用配置详解
2017/07/14 Javascript
前端防止用户重复提交js实现代码示例
2018/09/07 Javascript
JS实现音乐导航特效
2020/01/06 Javascript
JavaScript设计模式之观察者模式与发布订阅模式详解
2020/05/07 Javascript
[05:46]2018完美盛典-《同梦共竞》
2018/12/17 DOTA
python赋值操作方法分享
2013/03/23 Python
python 字符串split的用法分享
2013/03/23 Python
python实现人工智能Ai抠图功能
2019/09/05 Python
python根据时间获取周数代码实例
2019/09/30 Python
Python 列表推导式需要注意的地方
2020/10/23 Python
CSS3下的渐变文字效果实现示例
2018/03/02 HTML / CSS
世界领先的26岁以下学生和青少年旅行预订网站:StudentUniverse
2018/07/01 全球购物
德国综合购物网站:OTTO
2018/11/13 全球购物
Hotels.com韩国:海外国内旅行所需的酒店和住宿预订网站
2020/05/08 全球购物
产品销售员岗位职责
2013/12/18 职场文书
优秀本科生求职推荐信
2014/02/24 职场文书
最新大学生创业计划书写作攻略
2014/04/02 职场文书
初中作文评语大全
2014/04/23 职场文书
三月学雷锋月活动总结
2014/04/28 职场文书
党员教师群众路线对照检查材料思想汇报
2014/09/29 职场文书
2015年乡镇安全生产工作总结
2015/05/19 职场文书
只需要100行Python代码就可以实现的贪吃蛇小游戏
2021/05/27 Python
MySQL一些常用高级SQL语句
2021/07/03 MySQL