加载模型图_Tensorflow ckpt模型加载时的命名映射

tensorflow代码在加载ckpt模型时一般会使用如下代码：

saver = tf.train.Saver()
with tf.Session() as sess:saver.restore(sess, "model.ckpt")

那么问题来了，如果想把开源的ckpt模型加载到自己代码里不同的变量命名空间中，应该如何操作呢？

查看tensorflow文档可以发现，tf.train.Saver() 接受一个 var_list 参数，可以是一个 dict[str, SaveableObject]，即一个名字到tf变量的映射，利用它即可实现模型加载时的命名映射。

举一个栗子。在kinetics-i3d代码中，有如下一段

rgb_variable_map = {}
for variable in tf.global_variables():
if variable.name.split('/')[0] == 'RGB':if eval_type == 'rgb600':rgb_variable_map[variable.name.replace(':0', '')[len('RGB/inception_i3d/'):]] = variableelse:rgb_variable_map[variable.name.replace(':0', '')] = variablergb_saver = tf.train.Saver(var_list=rgb_variable_map, reshape=True)

功能是根据不同eval_type加载模型的时候，如果eval_type == 'rgb600'，就需要把待加载模型中的每个权重的名字都映射到计算图中加了'RGB/inception_i3d/'前缀的名字的变量。这样就实现了命名映射。

更多精彩内容，请滑至顶部点击右上角关注小宅哦~

作者：dailydreamer

加载模型图_Tensorflow ckpt模型加载时的命名映射相关推荐

tensorflow打印模型图_tensorflow实现打印pb模型的所有节点
{"moduleinfo":{"card_count":[{"count_phone":1,"count":1}],&q ...
tensorflow打印模型图_tensorflow 实现打印pb模型的所有节点
{"moduleinfo":{"card_count":[{"count_phone":1,"count":1}],&q ...
android加载长图宽度自适应,Android使用Glide加载超长图
app项目开发过程中,大都有查看图片详情的功能模块.实现的方式一般都是通过viewpager+photoView+Glide的方式.这种方式下,如果碰到加载超长图,比如说高清版的清明上河图/世界地图, ...
webview加载超长图
前言:android开发要经常和图片打交道.关于如何缓存,如何加载网上有很多框架.但是当我加载640*6149的长图时,实现起来效果并不理想,我用的是fresco加载.由于OpenGL渲染系统限制,A ...
UML 的九种模型图
1. UML的模型图 UML的模型图能够将被建模的系统的某一个方面的某一部分以图形的方式表示出来,不同的视图通过将多个不同的模型图有机组合在一起就能够描述系统模型的某方面的特征.UML的模型图 ...
ER模型详解-陈氏模型
转载自: http://blog.163.com/magicc_love/blog/static/18585366220142125836878/ Entity Relationship Model ...
打分法收益模型回归法风险模型最简明分析实测
本文的主要内容为,分别测试多因子模型中的风险模型和收益模型,并比较二者的绩效. 风险模型概述本文风险模型的自变量计算规则类似于 Fama-French 三因子模型的计算方法,模型当中的自变量为做多因 ...
针孔相机模型和相机镜头畸变模型
针孔相机模型和相机镜头畸变模型一.针孔相机模型图1. 针孔相机模型针孔相机模型(如图1所示)存在四个坐标系:世界坐标系.摄像机坐标系.图像物理坐标系和图像像素坐标系.假设现实世界的空间点的世界坐 ...
python加载模型文件进行图片分类_tensorflow通过模型文件,使用tensorboard查看其模型图Graph方式...
Google提供了一个工具,TensorBoard,它能以图表的方式分析你在训练过程中汇总的各种数据,其中包括Graph结构. 所以我们可以简单的写几行Pyhton,加载Graph,只在logdir里 ...

加载模型图_Tensorflow ckpt模型加载时的命名映射

加载模型图_Tensorflow ckpt模型加载时的命名映射相关推荐

最新文章

热门文章