Tensorflow: 从checkpoint文件中读取tensor方式
作者：noirblack 发布时间：2022-07-20 11:39:29
标签：Tensorflow,checkpoint,tensor
在使用pre-train model时候，我们需要restore variables from checkpoint files.
经常出现在checkpoint 中找不到”Tensor name not found”.
这时候需要查看一下ckpt中到底有哪些变量
import os
from tensorflow.python import pywrap_tensorflow
checkpoint_path = os.path.join(model_dir, "model.ckpt")
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
print("tensor_name: ", key)
print(reader.get_tensor(key))
可以显示ckpt中的tensor名字和值，当然也可以用pycharm调试。
来源：https://blog.csdn.net/noirblack/article/details/71430790
0
投稿猜你喜欢
- 本文实例讲述了php以post形式发送xml的方法。分享给大家供大家参考。具体方法如下：方法一，使用curl：$xml_data = <
最近小编遇到一个奇葩问题，就是上传代码时拷贝vs里面的代码不能直接粘贴，否则空格会不符合要求，怎么解决此问题呢？下面小编给大家分享我的解决方- 原来的题目设想为界面视觉效果的统一性，但是“统一”这个词似乎有点敏感，怕触动萌点无数，而我也无意去设定一个什么什么的统一性来侃侃而谈，极为专
- swoole —— 重新定义PHPswoole 的进程之间有两种通信方式，一种是消息队列(queue)，另一种是管道(pipe)，对swoo
- #!/usr/bin/env python# -*- coding: utf8 -*-import MySQLdbimport timeim
- 学在前面从本篇博客起，我们将实际完成几个小案例，第一个就是银行卡号识别，预计本案例将写 5 篇左右的博客才可以完成，一起加油吧。本文的目标是
- 离散特征的编码分为两种情况： 1、离散特征的取值之间没有大小的意义，比如color：[red,blue],那么就使用one-hot编码2、离
- 1.文件的读取和显示方法1： f=open(r'G:\2.txt') print f.read()&nb
- 补充说明，外键：不要使用外键，一切外键概念都在应用层解决。补充说明，数据库的列，也就是字段名，尽量带上飘符号`数据库存在的意义：数据存储和数
- 添加表字段alter table table1 add transactor varchar(10) not Null;alter tabl
- 使用到的库好多人都想了解一下如何对摄像头进行调用，然后进行目标检测，于是我做了这个小BLOG。opencv-python==4.1.2.30
- LNMP一键包安装的Mysql默认开启Mysql日志，如果网站对数据库读写较为频繁的话可能会产生大量日志，并占用磁盘空间，还有可
- 摘要：下拉菜单经常带来更多的可用性问题，并且常常容易被混淆。这是因为网页设计师们往往会在不同的几种情况下使用它。同时，滚动的菜单降低了网页的
- 1_cookie[掌握]解释: 用来保持服务器和浏览器交互的状态的, 由服务器设置,存储在浏览器作用: 用来做广告推送cookie的设置和获
- 只添加了一些自己想到的常用的功能,欢迎大家补充添加自己的好的思路. 通用的正则和方法可以写在RegExpObj中,
- HTML代码：在一个div容器内，设置3个span<body><div id="i1"> &nb
- 两者在使用效果上没有任何区别，都是为了将实例化后的vue挂载到指定的dom元素中。如果在实例化vue的时候指定el，则该vue将会渲染在此e
- 1.where语法和用法(1)语法：where <criteria> 即where <查询条件>具体查询语句：sel
- 在windows中罗技K380可以安装Logitech Options来实现这个Fn锁定功能。在linux中如何实现Logitech Opt
- 前言前段时间，因为项目需求，需要根据关键词搜索聊天记录，这不就是一个搜索引擎的功能吗？于是我第一时间想到的就是 ElasticSearch
