tensorflow选择性读取权重
来源:互联网 发布:电脑网络摄像头 编辑:程序博客网 时间:2024/05/21 16:17
在使用tensorflow加载网络权重的时候,直接使用tf.train.Saver().restore(sess, ‘ckpt’)的话是直接加载了全部权重,我们可能只需要加载网络的前几层权重,或者只要或者不要特定几层的权重,这时可以使用下面的方法:
var = tf.global_variables()var_to_restore = [val for val in var if 'conv1' in val.name or 'conv2'in val.name]saver = tf.train.Saver(var_to_restore )saver.restore(sess, os.path.join(model_dir, model_name))var_to_init = [val for val in var if 'conv1' not in val.name or 'conv2'not in val.name]tf.initialize_variables(var_to_init)
这样就只从ckpt文件里只读取到了两层卷积的卷积参数,前提是你的前两层网络结构和名字和ckpt文件里定义的一样。将var_to_restore和var_to_init反过来就是加载名字中不包含conv1、2的权重。
如果使用tensorflow的slim选择性读取权重的话就更方便了
exclude = ['layer1', 'layer2']variables_to_restore = slim.get_variables_to_restore(exclude=exclude)saver = tf.train.Saver(variables_to_restore)saver.restore(sess, os.path.join(model_dir, model_name))
这样就完成了不读取ckpt文件中’layer1’, ‘layer2’权重
阅读全文
1 0
- tensorflow选择性读取权重
- 【tensorflow】选择性读取加载权重,fine-tuning
- 【tensorflow】fine-tuning, 选择性加载ckpt部分权重
- tensorflow训练权重的保存和读取
- tensorflow 权重初始化
- tensorflow存储图和训练的权重为.pb,然后读取.pb并使用
- tensorflow/cifar10.py权重损失
- Tensorflow 权重衰减的使用
- tensorflow保存图和权重
- 根据权重随机读取数据
- tensorflow 输出权重 到csv或txt
- tensorflow输出权重值和偏差
- tensorflow爬坑行:数据读取
- Tensorflow图片数据读取
- Tensorflow读取数据
- Tensorflow | 读取csv文件
- Tensorflow直接读取二进制文件
- Tensorflow读取数据
- 使用raise语句自己触发异常
- (二)文件I/O 3.creat函数和close函数
- LwIP动态内存管理
- mathematica StringCases和StringReplace,Regex
- 【深度学习】初始化笔记
- tensorflow选择性读取权重
- LintCode-分治-验证二叉查找树
- 《Eve: Valkyrie Warzone》发布,非VR玩家也能公平开战
- dubbo的consumer代理的生成方式源码
- Tomcat服务器
- Jzoj3780 Magical GCD
- kubectl常用命令
- jsoncpp fatal error C1083: 无法打开编译器生成的文件 No such file or
- YJBannerView 的使用方法详解