tensorflow獲取預(yù)訓(xùn)練模型某層參數(shù)并賦值到當(dāng)前網(wǎng)絡(luò)指定層方式
已經(jīng)有了一個(gè)預(yù)訓(xùn)練的模型，我需要從其中取出某一層，把該層的weights和biases賦值到新的網(wǎng)絡(luò)結(jié)構(gòu)中，可以使用tensorflow中的pywrap_tensorflow(用來(lái)讀取預(yù)訓(xùn)練模型的參數(shù)值)結(jié)合Session.assign()進(jìn)行操作。
這種需求即預(yù)訓(xùn)練模型可能為單分支網(wǎng)絡(luò)，當(dāng)前網(wǎng)絡(luò)為多分支，我需要把單分支A復(fù)用到到多個(gè)分支去（B，C，D）。
先導(dǎo)入對(duì)應(yīng)的工具包
from tensorflow.python import pywrap_tensorflow
接下來(lái)的操作在一個(gè)tf.Session中進(jìn)行
reader = pywrap_tensorflow.NewCheckpointReader(pre_train_model_path)
# 獲取當(dāng)前圖可訓(xùn)練變量
trainable_variables = tf.trainable_variables()
# 需要賦值的當(dāng)前網(wǎng)絡(luò)層變量，這里只是隨便起的名字。
restore_v_target_name = "fc_target"
# 需要的預(yù)訓(xùn)練模型中的某層的名字
restore_v_source_name = "fc_source"
for v in trainable_variables:
if restore_v_target_name == v.name:
# 回復(fù)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獲取預(yù)訓(xùn)練模型某層參數(shù)并賦值到當(dāng)前網(wǎng)絡(luò)指定層方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
利用Python中?Rembg庫(kù)實(shí)現(xiàn)去除圖片背景
這篇文章主要介紹了利用Python中?Rembg庫(kù)實(shí)現(xiàn)去除圖片背景，文章基于?Rembg庫(kù)得運(yùn)用展開(kāi)詳細(xì)介紹，具有一定的參考價(jià)值，需要的小伙伴可以參考一下2022-05-05
作為一個(gè)運(yùn)維人員，經(jīng)常需要獲取系統(tǒng)的的各種信息，使用python會(huì)很方便幫助獲得，這篇文章運(yùn)用實(shí)例告訴大家如何利用Python來(lái)獲取操作系統(tǒng)的信息，有需要的可以參考借鑒。2016-09-09
Python中實(shí)現(xiàn)地圖可視化的方法小結(jié)
Python提供了多個(gè)強(qiáng)大的庫(kù),如Folium、Matplotlib、Geopandas等,使得創(chuàng)建漂亮而具有信息量的地圖變得簡(jiǎn)單而靈活,本文將詳細(xì)介紹如何使用這些庫(kù)繪制漂亮的地圖,感興趣的可以了解下2023-12-12
Pygame實(shí)戰(zhàn)練習(xí)之保護(hù)單身狗游戲
下面這篇文章主要給大家介紹了關(guān)于如何利用python寫一個(gè)簡(jiǎn)單的由經(jīng)典躲避類益智小游戲修改的保護(hù)單身狗游戲的相關(guān)資料,需要的朋友可以參考下2021-09-09
python list count統(tǒng)計(jì)個(gè)數(shù)的實(shí)現(xiàn)
這篇文章主要介紹了python list count統(tǒng)計(jì)個(gè)數(shù)的實(shí)現(xiàn)方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-02-02
使用Python實(shí)現(xiàn)不同需求的排行榜功能
這篇文章主要為大家介紹了Python實(shí)現(xiàn)不同需求的排行榜功能,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2024-01-01
python繪制子圖技巧之plt.subplot、plt.subplots及坐標(biāo)軸修改
一個(gè)圖片里邊繪制多個(gè)圖像是繪圖中的常見(jiàn)需求,下面這篇文章主要給大家介紹了關(guān)于python繪制子圖技巧之plt.subplot、plt.subplots及坐標(biāo)軸修改的相關(guān)資料,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-05-05
詳解python __init__.py 和 __all__作用
導(dǎo)入文件夾包的時(shí)候，會(huì)運(yùn)行寫在該文件夾包下的__init__.py文件，這主要是__init__.py的作用，本文結(jié)合示例代碼介紹了python __init__.py 和 __all__作用，感興趣的朋友一起看看吧2023-02-02
SecureCRTSecure7密碼查看的方法大概可以分為兩個(gè)步驟，第一步需要查看系統(tǒng)保存的連接的ini文件，第二步破解加密之后的密碼，具體腳本請(qǐng)參考下本文2021-06-06
Python實(shí)戰(zhàn)購(gòu)物車項(xiàng)目的實(shí)現(xiàn)參考
今天小編就為大家分享一篇關(guān)于Python實(shí)戰(zhàn)購(gòu)物車項(xiàng)目的實(shí)現(xiàn)參考，小編覺(jué)得內(nèi)容挺不錯(cuò)的，現(xiàn)在分享給大家，具有很好的參考價(jià)值，需要的朋友一起跟隨小編來(lái)看看吧2019-02-02