Tensorflow讀取并輸出已保存模型的權(quán)重?cái)?shù)值方式
更新時(shí)間：2020年01月04日 11:16:32 作者：AManFromEarth
今天小編就為大家分享一篇Tensorflow讀取并輸出已保存模型的權(quán)重?cái)?shù)值方式，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看不看
這篇文章是為了對網(wǎng)絡(luò)模型的權(quán)重輸出，可以用來轉(zhuǎn)換成其他框架的模型。
import tensorflow as tf
from tensorflow.python import pywrap_tensorflow
#首先，使用tensorflow自帶的python打包庫讀取模型
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ù)值方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章介紹了Python使用pyecharts控件繪制圖表的方法，文中通過示例代碼介紹的非常詳細(xì)。對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2022-06-06
這篇文章主要介紹了Python圖像處理二值化方法實(shí)例匯總,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-07-07
python 使用 requests 模塊發(fā)送http請求 的方法
本文分步驟給大家介紹了python 使用 requests 模塊發(fā)送http請求 的方法,非常不錯(cuò),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2018-12-12
這篇文章主要介紹了簡單了解pytest測試框架setup和tearDown,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-04-04
Python腳本簡單實(shí)現(xiàn)打開默認(rèn)瀏覽器登錄人人和打開QQ的方法
這篇文章主要介紹了Python腳本簡單實(shí)現(xiàn)打開默認(rèn)瀏覽器登錄人人和打開QQ的方法,涉及Python針對瀏覽器及應(yīng)用程序的相關(guān)操作技巧,代碼非常簡單實(shí)用,需要的朋友可以參考下2016-04-04
flask數(shù)據(jù)庫序列化操作實(shí)例探究
這篇文章主要為大家介紹了flask數(shù)據(jù)庫序列化操作實(shí)例探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2024-01-01
Django在pycharm下修改默認(rèn)啟動(dòng)端口的方法
今天小編就為大家分享一篇Django在pycharm下修改默認(rèn)啟動(dòng)端口的方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
django多個(gè)APP的urls設(shè)置方法(views重復(fù)問題解決)
今天小編就為大家分享一篇django多個(gè)APP的urls設(shè)置方法(views重復(fù)問題解決)，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
將Python的Django框架與認(rèn)證系統(tǒng)整合的方法
這篇文章主要介紹了將Python的Django框架與認(rèn)證系統(tǒng)整合的方法,包括指定認(rèn)證后臺(tái)和編寫認(rèn)證后臺(tái)等內(nèi)容,需要的朋友可以參考下2015-07-07
這篇文章主要介紹了OpenCV圖像處理中常用的幾個(gè)圖像幾何變換：裁剪、放大、縮小、平移、錯(cuò)切、鏡像、旋轉(zhuǎn)、透視等。文中示例代碼非常詳細(xì)，需要的朋友可以參考一下2021-12-12