Tensorflow讀取并輸出已保存模型的權(quán)重?cái)?shù)值方式
更新時(shí)間：2020年01月04日 11:16:32 作者：AManFromEarth
今天小編就為大家分享一篇Tensorflow讀取并輸出已保存模型的權(quán)重?cái)?shù)值方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看不看
這篇文章是為了對(duì)網(wǎng)絡(luò)模型的權(quán)重輸出，可以用來(lái)轉(zhuǎn)換成其他框架的模型。
import tensorflow as tf
from tensorflow.python import pywrap_tensorflow
#首先，使用tensorflow自帶的python打包庫(kù)讀取模型
model_reader = pywrap_tensorflow.NewCheckpointReader(r"model.ckpt")
#然后，使reader變換成類似于dict形式的數(shù)據(jù)
var_dict = model_reader.get_variable_to_shape_map()
#最后，循環(huán)打印輸出
for key in var_dict:
print("variable name: ", key)
print(model_reader.get_tensor(key))
結(jié)果（其中一個(gè)權(quán)重的示例）
........ variable name: InceptionV3/Mixed_7c/Branch_3/Conv2d_0b_1x1/weights [[[[ 0.00013783 -0.00251428 0.02235526 ... -0.01409702 0.00340105 -0.00752808] [ 0.01590012 -0.00258413 -0.00627338 ... -0.03600493 0.01220086 -0.01254225] [-0.02394262 -0.00764508 -0.00895328 ... -0.01731405 0.03568469 0.00918952] ... [-0.01865693 -0.00358359 -0.02342288 ... 0.00935838 0.00367858 -0.00976252] [ 0.01297642 0.00223457 0.00652326 ... -0.00762609 -0.0136022 -0.01129473] [-0.01395879 -0.00920246 0.01061887 ... 0.0236958 0.00041993 -0.01291134]]]] ......
以上這篇Tensorflow讀取并輸出已保存模型的權(quán)重?cái)?shù)值方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Django集成MongoDB實(shí)現(xiàn)過(guò)程解析
這篇文章主要介紹了Django集成MongoDB實(shí)現(xiàn)過(guò)程解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-12-12
python讀取excel進(jìn)行遍歷/xlrd模塊操作
這篇文章主要介紹了python讀取excel進(jìn)行遍歷/xlrd模塊操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-07-07
如何利用python實(shí)現(xiàn)Simhash算法
這篇文章主要介紹了如何利用python實(shí)現(xiàn)Simhash算法，文章基于python的相關(guān)資料展開(kāi)Simhash算法的詳細(xì)介紹，具有一定的參考價(jià)值，感興趣的小伙伴可以參考一下2022-06-06
今天小編就為大家分享一篇解決python大批量讀寫.doc文件的問(wèn)題。具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-05-05
python控制windows剪貼板,向剪貼板中寫入圖片的實(shí)例
今天小編就為大家分享一篇python控制windows剪貼板,向剪貼板中寫入圖片的實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-05-05
這篇文章主要介紹了微軟官方的Python插件，已經(jīng)自帶很多功能，下面是插件功能描述，其中部分內(nèi)容我做了翻譯,需要的朋友可以參考下2018-04-04
舉例講解Python面相對(duì)象編程中對(duì)象的屬性與類的方法
這篇文章主要介紹了Python面相對(duì)象編程中對(duì)象的屬性與類的方法,類和對(duì)象以及方法這些都是面相對(duì)象的編程語(yǔ)言所具有的基礎(chǔ)內(nèi)容,需要的朋友可以參考下2016-01-01
Python數(shù)據(jù)結(jié)構(gòu)與算法之列表（鏈表，linked list）簡(jiǎn)單實(shí)現(xiàn)
這篇文章主要介紹了Python數(shù)據(jù)結(jié)構(gòu)與算法之列表（鏈表，linked list）簡(jiǎn)單實(shí)現(xiàn)，具有一定參考價(jià)值，需要的朋友可以了解下。2017-10-10
這篇文章主要介紹了Python3壓縮和解壓縮實(shí)現(xiàn)代碼,代碼簡(jiǎn)單易懂，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-03-03
Python+OpenCV實(shí)戰(zhàn)之實(shí)現(xiàn)文檔掃描
這篇文章主要為大家詳細(xì)介紹了Python+Opencv如何實(shí)現(xiàn)文檔掃描的功能，文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2022-09-09