Tensorflow读取并输出已保存模型的权重数值方式

更新时间：2020年01月04日 11:16:32 作者：AManFromEarth

今天小编就为大家分享一篇Tensorflow读取并输出已保存模型的权重数值方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看不看

这篇文章是为了对网络模型的权重输出，可以用来转换成其他框架的模型。

import tensorflow as tf
from tensorflow.python import pywrap_tensorflow

#首先，使用tensorflow自带的python打包库读取模型
model_reader = pywrap_tensorflow.NewCheckpointReader(r"model.ckpt")

#然后，使reader变换成类似于dict形式的数据
var_dict = model_reader.get_variable_to_shape_map()

#最后，循环打印输出
for key in var_dict:
  print("variable name: ", key)
  print(model_reader.get_tensor(key))

结果（其中一个权重的示例）

........
variable name: InceptionV3/Mixed_7c/Branch_3/Conv2d_0b_1x1/weights
[[[[ 0.00013783 -0.00251428 0.02235526 ... -0.01409702 0.00340105
  -0.00752808]
  [ 0.01590012 -0.00258413 -0.00627338 ... -0.03600493 0.01220086
  -0.01254225]
  [-0.02394262 -0.00764508 -0.00895328 ... -0.01731405 0.03568469
   0.00918952]
  ...
  [-0.01865693 -0.00358359 -0.02342288 ... 0.00935838 0.00367858
  -0.00976252]
  [ 0.01297642 0.00223457 0.00652326 ... -0.00762609 -0.0136022
  -0.01129473]
  [-0.01395879 -0.00920246 0.01061887 ... 0.0236958  0.00041993
  -0.01291134]]]]
......

以上这篇Tensorflow读取并输出已保存模型的权重数值方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

利用Python中unittest实现简单的单元测试实例详解
如果项目复杂，进行单元测试是保证降低出错率的好方法，Python提供的unittest可以很方便的实现单元测试，从而可以替换掉繁琐杂乱的main函数测试的方法，将测试用例、测试方法进行统一的管理和维护。本文主要介绍了利用Python中unittest实现简单的单元测试。
2017-01-01
Python getattr()函数使用方法代码实例
这篇文章主要介绍了Python getattr()函数使用方法代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-08-08
python3定位并识别图片验证码实现自动登录功能
这篇文章主要介绍了python3定位并识别图片验证码实现自动登录功能,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-01-01
使用tensorflow实现VGG网络,训练mnist数据集方式
这篇文章主要介绍了使用tensorflow实现VGG网络,训练mnist数据集方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-05-05
Python 连接 MySQL 的几种方法
这篇文章主要介绍了Python 连接 MySQL 的几种方法，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下
2020-09-09
python中的class_static的@classmethod的巧妙用法
python中的class_static的@classmethod的使用 classmethod的使用，主要针对的是类而不是对象，在定义类的时候往往会定义一些静态的私有属性，今天通过示例代码看下classmethod的妙用
2021-06-06
Python字符串逆序的实现方法【一题多解】
今天小编就为大家分享一篇关于Python字符串逆序的实现方法【一题多解】，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧
2019-02-02
python循环控制之break和continue流程控制语句
这篇文章主要介绍了python循环控制之break流程控制语句，Python中提供了两个关键字用来控制循环语句,分别是break和continue，本文都有介绍,需要的朋友可以参考一下
2022-03-03
python命令行引导用户填写ssh登录信息详解
这篇文章主要为大家介绍了python命令行引导用户填写ssh登录信息详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2023-11-11
python实现单链表的方法示例
这篇文章主要给大家介绍了关于python实现单链表的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用python具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧
2019-09-09