tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式


Posted in Python onJanuary 24, 2020

已经有了一个预训练的模型,我需要从其中取出某一层,把该层的weights和biases赋值到新的网络结构中,可以使用tensorflow中的pywrap_tensorflow(用来读取预训练模型的参数值)结合Session.assign()进行操作。

这种需求即预训练模型可能为单分支网络,当前网络为多分支,我需要把单分支A复用到到多个分支去(B,C,D)。

tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式

先导入对应的工具包

from tensorflow.python import pywrap_tensorflow

接下来的操作在一个tf.Session中进行

reader = pywrap_tensorflow.NewCheckpointReader(pre_train_model_path)

# 获取当前图可训练变量
trainable_variables = tf.trainable_variables()
# 需要赋值的当前网络层变量,这里只是随便起的名字。
restore_v_target_name = "fc_target"
# 需要的预训练模型中的某层的名字
restore_v_source_name = "fc_source"
for v in trainable_variables:
  if restore_v_target_name == v.name:
   # 回复weights和biases
    sess.run(
      tf.assign(v, reader.get_tensor(restore_v_source_name + "/weights"))) if "weights" in v.name else sess.run(
      tf.assign(v, reader.get_tensor(restore_v_source_name + "/biases")))

以上这篇tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python字符串处理之count()方法的使用
May 18 Python
Python实现控制台中的进度条功能代码
Dec 22 Python
Python获取指定字符前面的所有字符方法
May 02 Python
python 读取目录下csv文件并绘制曲线v111的方法
Jul 06 Python
python 读取文本文件的行数据,文件.splitlines()的方法
Jul 12 Python
django多对多表的创建,级联删除及手动创建第三张表
Jul 25 Python
Python学习笔记之While循环用法分析
Aug 14 Python
简单了解django文件下载方式
Feb 10 Python
python 错误处理 assert详解
Apr 20 Python
python boto和boto3操作bucket的示例
Oct 30 Python
python 图像增强算法实现详解
Jan 24 Python
在NumPy中深拷贝和浅拷贝相关操作的定义和背后的原理
Apr 14 Python
keras获得某一层或者某层权重的输出实例
Jan 24 #Python
浅谈keras的深度模型训练过程及结果记录方式
Jan 24 #Python
关于Keras模型可视化教程及关键问题的解决
Jan 24 #Python
基于keras 模型、结构、权重保存的实现
Jan 24 #Python
Python 文件数据读写的具体实现
Jan 24 #Python
利用keras加载训练好的.H5文件,并实现预测图片
Jan 24 #Python
keras模型可视化,层可视化及kernel可视化实例
Jan 24 #Python
You might like
PHP cURL初始化和执行方法入门级代码
2015/05/28 PHP
初识通用数据库操作类――前端easyui-datagrid,form(php)
2015/07/31 PHP
基础的WordPress插件制作教程
2015/11/24 PHP
WordPress中用于检索模版的相关PHP函数使用解析
2015/12/15 PHP
PHPExcel中文帮助手册|PHPExcel使用方法(分享)
2017/06/09 PHP
PHP Post获取不到非表单数据的问题解决办法
2018/02/27 PHP
PHP实现的函数重载功能示例
2018/08/03 PHP
JavaScript 变量命名规则
2009/09/23 Javascript
浅谈Javascript事件处理程序的几种方式
2012/06/27 Javascript
JQueryiframe页面操作父页面中的元素与方法(实例讲解)
2013/11/19 Javascript
验证码在IE中不刷新而谷歌等浏览器正常的解决方案
2014/03/18 Javascript
Javascript 拖拽雏形(逐行分析代码,让你轻松了拖拽的原理)
2015/01/23 Javascript
前端性能优化及技巧
2016/05/06 Javascript
jquery实现点击弹出可放大居中及关闭的对话框(附demo源码下载)
2016/05/10 Javascript
在JSP中如何实现MD5加密的方法
2016/11/02 Javascript
基于vue.js实现侧边菜单栏
2017/03/20 Javascript
浅谈angular2路由预加载策略
2017/10/04 Javascript
利用node 判断打开的是文件 还是 文件夹的实例
2019/06/10 Javascript
Vue formData实现图片上传
2019/08/20 Javascript
JavaScript canvas绘制渐变颜色的矩形
2020/02/18 Javascript
python开发之文件操作用法实例
2015/11/13 Python
EM算法的python实现的方法步骤
2018/01/02 Python
django项目环境搭建及在虚拟机本地创建django项目的教程
2019/08/02 Python
Python csv模块使用方法代码实例
2019/08/29 Python
浅谈numpy中np.array()与np.asarray的区别以及.tolist
2020/06/03 Python
Pycharm2020最新激活码|永久激活(附最新激活码和插件的详细教程)
2020/09/29 Python
django使用channels实现通信的示例
2020/10/19 Python
印度尼西亚手表和包包商店:Urban Icon
2019/12/12 全球购物
总经理助理的八要求
2013/11/12 职场文书
媒体宣传策划方案
2014/05/25 职场文书
本科毕业论文答辩稿
2015/06/23 职场文书
施工现场安全管理制度
2015/08/05 职场文书
高中英语教学反思范文
2016/03/02 职场文书
Python预测分词的实现
2021/06/18 Python
Django路由层如何获取正确的url
2021/07/15 Python
浅谈redis整数集为什么不能降级
2021/07/25 Redis