将tensorflow的ckpt模型存储为npy的实例


Posted in Python onJuly 09, 2018

实例如下所示:

#coding=gbk
import numpy as np
import tensorflow as tf
from tensorflow.python import pywrap_tensorflow

checkpoint_path='model.ckpt-5000'#your ckpt path
reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map=reader.get_variable_to_shape_map()

alexnet={}
alexnet_layer = ['conv1','conv2','conv3','conv4','conv5','fc6','fc7','fc8']
add_info = ['weights','biases']

alexnet={'conv1':[[],[]],'conv2':[[],[]],'conv3':[[],[]],'conv4':[[],[]],'conv5':[[],[]],'fc6':[[],[]],'fc7':[[],[]],'fc8':[[],[]]}


for key in var_to_shape_map:
 #print ("tensor_name",key)

 str_name = key
 # 因为模型使用Adam算法优化的,在生成的ckpt中,有Adam后缀的tensor
 if str_name.find('Adam') > -1:
  continue

 print('tensor_name:' , str_name)

 if str_name.find('/') > -1:
  names = str_name.split('/')
  # first layer name and weight, bias
  layer_name = names[0]
  layer_add_info = names[1]
 else:
  layer_name = str_name
  layer_add_info = None

 if layer_add_info == 'weights':
  alexnet[layer_name][0]=reader.get_tensor(key)
 elif layer_add_info == 'biases':
  alexnet[layer_name][1] = reader.get_tensor(key)
 else:
  alexnet[layer_name] = reader.get_tensor(key)

# save npy
np.save('alexnet_pointing04.npy',alexnet)
print('save npy over...')
#print(alexnet['conv1'][0].shape)
#print(alexnet['conv1'][1].shape)

以上这篇将tensorflow的ckpt模型存储为npy的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python 开发Activex组件方法
Nov 08 Python
举例讲解Django中数据模型访问外键值的方法
Jul 21 Python
python+pygame简单画板实现代码实例
Dec 13 Python
python运行时强制刷新缓冲区的方法
Jan 14 Python
python获取地震信息 微信实时推送
Jun 18 Python
使用OpenCV实现仿射变换—平移功能
Aug 29 Python
python程序 线程队列queue使用方法解析
Sep 23 Python
关于pandas的离散化,面元划分详解
Nov 22 Python
django 链接多个数据库 并使用原生sql实现
Mar 28 Python
Visual Studio code 配置Python开发环境
Sep 11 Python
python实现视频压缩功能
Dec 18 Python
Python使用tkinter制作在线翻译软件
Feb 22 Python
caffe binaryproto 与 npy相互转换的实例讲解
Jul 09 #Python
利用Python如何生成便签图片详解
Jul 09 #Python
Caffe均值文件mean.binaryproto转mean.npy的方法
Jul 09 #Python
使用Python更换外网IP的方法
Jul 09 #Python
使用Python AIML搭建聊天机器人的方法示例
Jul 09 #Python
Python迭代器与生成器用法实例分析
Jul 09 #Python
numpy的文件存储.npy .npz 文件详解
Jul 09 #Python
You might like
《魔兽世界》惊魂幻象将获得调整
2020/03/08 其他游戏
服务器上配置PHP运行环境教程
2015/02/12 PHP
浅谈PHP中的
2016/04/23 PHP
PHP实现的观察者模式实例
2017/06/21 PHP
Laravel路由研究之domain解决多域名问题的方法示例
2019/04/04 PHP
JavaScript去除空格的几种方法
2006/10/03 Javascript
JS自动缩小超出大小的图片
2012/10/12 Javascript
Web跨浏览器进程通信(Web跨域)
2013/04/17 Javascript
纯文字版返回顶端的js代码
2013/08/01 Javascript
js获取多个tagname的节点数组
2013/09/22 Javascript
jquery简单实现滚动条下拉DIV固定在头部不动
2013/11/25 Javascript
javascript运行机制之this详细介绍
2014/02/07 Javascript
JQuery以JSON方式提交数据到服务端示例代码
2014/05/05 Javascript
JS中call/apply、arguments、undefined/null方法详解
2016/02/15 Javascript
在JavaScript中模拟类(class)及类的继承关系
2016/05/20 Javascript
vue.js指令v-for使用及索引获取
2016/11/03 Javascript
js实现日期显示的一些操作(实例讲解)
2017/07/27 Javascript
jQuery实现导航样式布局操作示例【可自定义样式布局】
2018/07/24 jQuery
详解Element 指令clickoutside源码分析
2019/02/15 Javascript
layui自定义插件citySelect实现省市区三级联动选择
2019/07/26 Javascript
python通过scapy获取局域网所有主机mac地址示例
2014/05/04 Python
python图像处理之反色实现方法
2015/05/30 Python
用python实现的线程池实例代码
2018/01/06 Python
PyCharm代码整体缩进,反向缩进的方法
2018/06/25 Python
python创建文件备份的脚本
2018/09/11 Python
Python列表与元组的异同详解
2019/07/02 Python
pytorch 指定gpu训练与多gpu并行训练示例
2019/12/31 Python
Tensorflow与Keras自适应使用显存方式
2020/06/22 Python
pycharm激活码2020最新分享适用pycharm2020最新版亲测可用
2020/11/22 Python
好的演讲稿开场白
2013/12/30 职场文书
公司部门司机岗位职责
2014/01/03 职场文书
主题婚礼策划方案
2014/02/10 职场文书
社区文化建设方案
2014/05/02 职场文书
离职保密承诺书
2014/05/28 职场文书
廉洁自律承诺书范文
2015/04/28 职场文书
Vue的过滤器你真了解吗
2022/02/24 Vue.js