tensorflow實現(xiàn)讀取模型中保存的值 tf.train.NewCheckpointReader
更新時間：2020年02月10日 11:10:06 作者：onexming
今天小編就為大家分享一篇tensorflow實現(xiàn)讀取模型中保存的值 tf.train.NewCheckpointReader，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
使用tf.trian.NewCheckpointReader(model_dir)
一個標準的模型文件有一下文件, model_dir就是MyModel(沒有后綴)
checkpoint Model.meta Model.data-00000-of-00001 Model.index
import tensorflow as tf
import pprint # 使用pprint 提高打印的可讀性
NewCheck =tf.train.NewCheckpointReader("model")
打印模型中的所有變量
print("debug_string:\n")
pprint.pprint(NewCheck.debug_string().decode("utf-8"))
其中有3個字段, 分別是名字, 數(shù)據(jù)類型, shape
獲取變量中的值
print("get_tensor:\n")
pprint.pprint(NewCheck.get_tensor("D/conv2d/bias"))
print("get_variable_to_dtype_map\n")
pprint.pprint(NewCheck.get_variable_to_dtype_map())
print("get_variable_to_shape_map\n")
pprint.pprint(NewCheck.get_variable_to_shape_map())
以上這篇tensorflow實現(xiàn)讀取模型中保存的值 tf.train.NewCheckpointReader就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
在開發(fā)過程中，配置文件是少不了的，而且配置文件是有專門的格式的，比如：ini,?yaml,?toml?等等。而對于?Python?而言，也都有相應(yīng)的庫來解析相應(yīng)格式的文件，下面我們來看看?ini?文件要如何解析2022-09-09
每次在讀或?qū)懳募r,獲取腳本文件的絕對路徑和相對路徑時很容易搞混,下面這篇文章主要給大家介紹了關(guān)于python文件讀取和導包的絕對路徑、相對路徑的相關(guān)資料,需要的朋友可以參考下2022-04-04
Python 中eval()函數(shù)的正確使用及其風險分析(使用示例)
eval()是一個功能強大的工具,但使用時必須非常小心,了解其工作原理和潛在的風險是確保安全使用的關(guān)鍵,通過遵循上述建議,可以在享受eval()帶來的便利的同時,最大限度地減少安全風險,本文介紹Python 中`eval()`函數(shù)的正確使用及其風險分析,感興趣的朋友一起看看吧2024-07-07
Python嵌套函數(shù)，作用域與偏函數(shù)用法實例分析
這篇文章主要介紹了Python嵌套函數(shù)，作用域與偏函數(shù)用法,結(jié)合實例形式分析了Python嵌套函數(shù)，作用域與偏函數(shù)的功能、定義與相關(guān)使用方法,需要的朋友可以參考下2019-12-12
Tensorflow的可視化工具Tensorboard的初步使用詳解
這篇文章主要介紹了Tensorflow的可視化工具Tensorboard的初步使用詳解，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-02-02
Python實現(xiàn)輕松合并doc為txt的示例代碼
這篇文章主要為大家詳細介紹了如何利用Python編程語言和wxPython模塊,打開指定文件夾中的DOC文檔,并將它們的內(nèi)容合并成一個便捷的TXT文檔,需要的可以參考下2024-03-03
這篇文章主要介紹了python 帶時區(qū)的日期格式化操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-10-10
django獲取from表單multiple-select的value和id的方法
今天小編就為大家分享一篇django獲取from表單multiple-select的value和id的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
今天小編就為大家分享一篇python取均勻不重復(fù)的隨機數(shù)方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-11-11
python裝飾器就是用于拓展原來函數(shù)功能的一種函數(shù)，這個函數(shù)的特殊之處在于它的返回值也是一個函數(shù)，使用python裝飾器的好處就是在不用更改原函數(shù)的代碼前提下給函數(shù)增加新的功能2021-07-07