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 相关文章推荐
Python3 入门教程 简单但比较不错
Nov 29 Python
简单掌握Python中glob模块查找文件路径的用法
Jul 05 Python
Python对象属性自动更新操作示例
Jun 15 Python
python3爬虫怎样构建请求header
Dec 23 Python
python SVD压缩图像的实现代码
Nov 05 Python
Python安装依赖(包)模块方法详解
Feb 14 Python
Python 使用 environs 库定义环境变量的方法
Feb 25 Python
python爬虫开发之Request模块从安装到详细使用方法与实例全解
Mar 09 Python
python 实现两个线程交替执行
May 02 Python
python相对企业语言优势在哪
Jun 12 Python
python连接mysql数据库并读取数据的实现
Sep 25 Python
Django利用elasticsearch(搜索引擎)实现搜索功能
Nov 26 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使用pclzip类实现文件压缩的方法(附pclzip类下载地址)
2016/04/30 PHP
什么是json和jsonp,jQuery json实例详详细说明
2012/12/11 Javascript
JavaScript实现的一个计算数字步数的算法分享
2014/12/06 Javascript
javascript实现禁止复制网页内容
2014/12/16 Javascript
JS响应鼠标点击实现两个滑块区间拖动效果
2015/10/26 Javascript
使用jQuery+EasyUI实现CheckBoxTree的级联选中特效
2015/12/06 Javascript
js实现兼容IE、Firefox的图片缩放代码
2015/12/08 Javascript
jQuery复合事件用法示例
2017/06/10 jQuery
详解jquery选择器的原理
2017/08/01 jQuery
基于ionic实现下拉刷新功能
2018/05/10 Javascript
Vue如何将页面导出成PDF文件
2020/08/17 Javascript
[00:32]2018DOTA2亚洲邀请赛OpTic出场
2018/04/03 DOTA
Python实现将HTML转换成doc格式文件的方法示例
2017/11/20 Python
django 按时间范围查询数据库实例代码
2018/02/11 Python
python Opencv将图片转为字符画
2021/02/19 Python
python 信息同时输出到控制台与文件的实例讲解
2018/05/11 Python
详解Django中类视图使用装饰器的方式
2018/08/12 Python
详解Python中的type和object
2018/08/15 Python
Python read函数按字节(字符)读取文件的实现
2019/07/03 Python
对Django中static(静态)文件详解以及{% static %}标签的使用方法
2019/07/28 Python
python3 字符串知识点学习笔记
2020/02/08 Python
python numpy实现多次循环读取文件 等间隔过滤数据示例
2020/03/14 Python
Python图像处理二值化方法实例汇总
2020/07/24 Python
Python控制鼠标键盘代码实例
2020/12/08 Python
CSS3制作半透明边框(Facebox)类似渐变
2012/12/09 HTML / CSS
HTML5 video标签(播放器)学习笔记(二):播放控制
2015/04/24 HTML / CSS
Expedia挪威官网:酒店、机票和租车
2018/03/03 全球购物
工程师岗位职责
2013/11/08 职场文书
丑小鸭教学反思
2014/02/03 职场文书
高中英语教学反思
2014/02/04 职场文书
历史专业大学生职业生涯规划书
2014/03/13 职场文书
国贸专业的职业规划书
2014/03/15 职场文书
教师作风整顿个人剖析材料
2014/10/10 职场文书
优秀团员事迹材料
2014/12/25 职场文书
2016年12月份红领巾广播稿
2015/12/21 职场文书
确保减税降费落地生根,用实实在在措施
2019/07/19 职场文书