将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 MySQL数据库连接池组件pymysqlpool详解
Jul 07 Python
python中的内置函数max()和min()及mas()函数的高级用法
Mar 29 Python
Python之批量创建文件的实例讲解
May 10 Python
对python 匹配字符串开头和结尾的方法详解
Oct 27 Python
Python将一个Excel拆分为多个Excel
Nov 07 Python
为什么Python中没有"a++"这种写法
Nov 27 Python
pycharm远程开发项目的实现步骤
Jan 20 Python
Python如何爬取实时变化的WebSocket数据的方法
Mar 09 Python
浅谈python多线程和多线程变量共享问题介绍
Apr 17 Python
Pytorch十九种损失函数的使用详解
Apr 29 Python
python随机模块random的22种函数(小结)
May 15 Python
Python 读写 Matlab Mat 格式数据的操作
May 19 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
PHP $_SERVER详解
2009/01/16 PHP
php下清空字符串中的HTML标签的代码
2010/09/06 PHP
php全排列递归算法代码
2012/10/09 PHP
使用Composer安装Yii框架的方法
2016/03/15 PHP
Zend Framework分发器用法示例
2016/12/11 PHP
JavaScript Date对象 日期获取函数
2010/12/19 Javascript
实现变速回到顶部的JavaScript代码
2011/05/09 Javascript
读jQuery之九 一些瑕疵说明
2011/06/21 Javascript
js Math 对象的方法
2013/09/01 Javascript
理解Javascript闭包
2013/11/01 Javascript
javascript自定义的addClass()方法
2014/05/28 Javascript
jQuery窗口、文档、网页各种高度的精确理解
2014/07/02 Javascript
jQuery学习笔记之2个小技巧
2015/01/19 Javascript
Jquery异步提交表单代码分享
2015/03/26 Javascript
面向切面编程(AOP)的理解
2015/05/01 Javascript
jQuery链式操作实例分析
2015/11/16 Javascript
jquery插件jquery.LightBox.js实现点击放大图片并左右点击切换效果(附demo源码下载)
2016/02/25 Javascript
BootStrap按钮标签及基本样式
2016/11/23 Javascript
jquery插件bootstrapValidator表单验证详解
2016/12/15 Javascript
Javascript 实现计算器时间功能详解及实例(二)
2017/01/08 Javascript
JavaScript获取键盘按键的键码(参照表)
2017/01/10 Javascript
微信小程序模版渲染详解
2018/01/26 Javascript
JS+HTML5本地存储Localstorage实现注册登录及验证功能示例
2020/02/10 Javascript
python zip文件 压缩
2008/12/24 Python
Python实现的knn算法示例
2018/06/14 Python
可能是最全面的 Python 字符串拼接总结【收藏】
2018/07/09 Python
Django JWT Token RestfulAPI用户认证详解
2019/01/23 Python
tornado+celery的简单使用详解
2019/12/21 Python
加拿大快时尚零售商:Ardene
2018/02/14 全球购物
雷蛇美国官网:Razer
2020/04/03 全球购物
一些Unix笔试题和面试题
2013/01/22 面试题
幼儿评语大全
2014/04/30 职场文书
社区娱乐活动方案
2014/08/21 职场文书
借名购房协议书范本
2014/10/06 职场文书
党员剖析材料范文
2014/12/18 职场文书
员工手册编写范本
2015/05/14 职场文书