Tensorflow 1.0之后模型文件、權(quán)重數(shù)值的讀取方式
更新時間：2020年02月12日 08:41:12 作者：地大大劉
今天小編就為大家分享一篇Tensorflow 1.0之后模型文件、權(quán)重數(shù)值的讀取方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
保存的文件有4個：
checkpoint
model-parameters.bin-46000.data-00000-of-00001
model-parameters.bin-46000.index
model-parameters.bin-46000.meta
讀取代碼：
from tensorflow.python import pywrap_tensorflow
with tf.Graph().as_default():
with tf.Session() as sess:
ckpt = tf.train.get_checkpoint_state('./checkpoint')
if ckpt and ckpt.model_checkpoint_path:
reader = pywrap_tensorflow.NewCheckpointReader('./checkpoint/model-parameters.bin-46000')
all_variables = reader.get_variable_to_shape_map()
w1 = reader.get_tensor("conv1/weight")
print(w1.shape)
print(w1)
print(all_variables)
else:
print('No checkpoint file found')
以上這篇Tensorflow 1.0之后模型文件、權(quán)重數(shù)值的讀取方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
TFRecord格式存儲數(shù)據(jù)與隊列讀取實例
今天小編就為大家分享一篇TFRecord格式存儲數(shù)據(jù)與隊列讀取實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01
這篇文章主要介紹了Python項目跨域問題解決方案,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-06-06
這篇文章主要介紹了一文詳述 Python 中的 property 語法,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-09-09
今天小編就為大家分享一篇python3中的md5加密實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
python3對拉勾數(shù)據(jù)進行可視化分析的方法詳解
這篇文章主要給大家介紹了關于python3對拉勾數(shù)據(jù)進行可視化分析的相關資料，文中通過示例代碼介紹的非常詳細，對大家學習或者使用Python3具有一定的參考學習價值，需要的朋友們下面來一起學習學習吧2019-04-04
Python讀取DataFrame的某行或某列的方法實現(xiàn)
Dataframe是Python中一種重要的數(shù)據(jù)處理工具,它能夠以表格形式存儲并處理數(shù)據(jù),本文主要介紹了Python讀取DataFrame的某行或某列的方法實現(xiàn),具有一定的參考價值,感興趣的可以了解一下2024-04-04
python微信聊天機器人改進版(定時或觸發(fā)抓取天氣預報、勵志語錄等,向好友推送)
這篇文章主要介紹了python微信聊天機器人改進版，定時或觸發(fā)抓取天氣預報、勵志語錄等，向好友推送 ,本文通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友可以參考下2019-04-04
這篇文章主要介紹了python實現(xiàn)屏保計時器的示例代碼，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-08-08
這篇文章主要為大家詳細介紹了Python tkinter實現(xiàn)日期選擇器，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-02-02
這篇文章主要介紹了python中numpy常用操作總結(jié)，NumPy是Python語言的一個擴充程序庫，支持大量高維度數(shù)組與矩陣運算，此外也針對數(shù)組運算提供大量的數(shù)學函數(shù)庫2022-09-09