tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式
更新时间：2020年01月24日 10:39:23 作者：曲草
今天小编就为大家分享一篇tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
已经有了一个预训练的模型，我需要从其中取出某一层，把该层的weights和biases赋值到新的网络结构中，可以使用tensorflow中的pywrap_tensorflow(用来读取预训练模型的参数值)结合Session.assign()进行操作。
这种需求即预训练模型可能为单分支网络，当前网络为多分支，我需要把单分支A复用到到多个分支去（B，C，D）。
先导入对应的工具包
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中Rembg库实现去除图片背景，文章基于Rembg库得运用展开详细介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-05-05
作为一个运维人员，经常需要获取系统的的各种信息，使用python会很方便帮助获得，这篇文章运用实例告诉大家如何利用Python来获取操作系统的信息，有需要的可以参考借鉴。2016-09-09
Python提供了多个强大的库,如Folium、Matplotlib、Geopandas等,使得创建漂亮而具有信息量的地图变得简单而灵活,本文将详细介绍如何使用这些库绘制漂亮的地图,感兴趣的可以了解下2023-12-12
下面这篇文章主要给大家介绍了关于如何利用python写一个简单的由经典躲避类益智小游戏修改的保护单身狗游戏的相关资料,需要的朋友可以参考下2021-09-09
这篇文章主要介绍了python list count统计个数的实现方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-02-02
这篇文章主要为大家介绍了Python实现不同需求的排行榜功能,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2024-01-01
python绘制子图技巧之plt.subplot、plt.subplots及坐标轴修改
一个图片里边绘制多个图像是绘图中的常见需求,下面这篇文章主要给大家介绍了关于python绘制子图技巧之plt.subplot、plt.subplots及坐标轴修改的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2022-05-05
详解python __init__.py 和 __all__作用
导入文件夹包的时候，会运行写在该文件夹包下的__init__.py文件，这主要是__init__.py的作用，本文结合示例代码介绍了python __init__.py 和 __all__作用，感兴趣的朋友一起看看吧2023-02-02
SecureCRTSecure7密码查看的方法大概可以分为两个步骤，第一步需要查看系统保存的连接的ini文件，第二步破解加密之后的密码，具体脚本请参考下本文2021-06-06
今天小编就为大家分享一篇关于Python实战购物车项目的实现参考，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧2019-02-02
最新评论