Tensorflow 1.0之后模型文件、权重数值的读取方式
更新时间：2020年02月12日 08:41:12 作者：地大大刘
今天小编就为大家分享一篇Tensorflow 1.0之后模型文件、权重数值的读取方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
保存的文件有4个：
checkpoint
model-parameters.bin-46000.data-00000-of-00001
model-parameters.bin-46000.index
model-parameters.bin-46000.meta
读取代码：
from tensorflow.python import pywrap_tensorflow
with tf.Graph().as_default():
with tf.Session() as sess:
ckpt = tf.train.get_checkpoint_state('./checkpoint')
if ckpt and ckpt.model_checkpoint_path:
reader = pywrap_tensorflow.NewCheckpointReader('./checkpoint/model-parameters.bin-46000')
all_variables = reader.get_variable_to_shape_map()
w1 = reader.get_tensor("conv1/weight")
print(w1.shape)
print(w1)
print(all_variables)
else:
print('No checkpoint file found')
以上这篇Tensorflow 1.0之后模型文件、权重数值的读取方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
基于spring boot 日志(logback)报错的解决方式
今天小编就为大家分享一篇基于spring boot 日志(logback)报错的解决方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
这篇文章主要介绍了在python中如何建立一个自己的包，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-02-02
Python 编码规范(Google Python Style Guide)
本项目并非 Google 官方项目, 而是由国内程序员凭热情创建和维护,对于想学习python的朋友可以参考一下2018-05-05
类与类方法是面向对象的编程语言中必不可少的特性,本文总结了Python面向对象编程中关于类和方法的学习笔记,需要的朋友可以参考下2016-06-06
tqdm是一个用于在Python中添加进度条的库，它可以很容易地集成到while循环中，这篇文章主要介绍了python循环打印进度条,需要的朋友可以参考下2023-05-05
本篇文章中我们给大家分享了关于Python实现登陆文件验证的方法和技巧，有兴趣的朋友们参考学习下。2018-10-10
python使用py2neo查询Neo4j的节点、关系及路径
本文介绍了使用Py2neo的NodeMatcher和RelationshipMatcher查询图中的节点和关系，以及通过执行Cypher语句的查询方式。感兴趣的小伙伴请看下文2021-08-08
这篇文章主要介绍了利用Python编写本地音乐播放器，主要以分享完整源码展开文章内容，具有一的参考价值，需要的小伙伴可以参考一下2022-03-03
这篇文章主要介绍了Python 异常处理的实例详解的相关资料,希望通过本文能帮助到大家掌握Python 异常的知识，需要的朋友可以参考下2017-09-09
牛顿迭代法又称为牛顿-拉夫逊（拉弗森）方法，它是牛顿在17世纪提出的一种在实数域和复数域上近似求解方程的方法。本文将利用图文详解牛顿迭代算法原理及实现，需要的可以参考一下2022-08-08
最新评论