tensorflow 實現(xiàn)從checkpoint中獲取graph信息
更新時間：2020年02月10日 15:13:02 作者：qq_28808697
今天小編就為大家分享一篇tensorflow 實現(xiàn)從checkpoint中獲取graph信息，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
代碼：
import tensorflow as tf
sess = tf.Session()
check_point_path = 'variables'
saver = tf.train.import_meta_graph('variables/save_variables.ckpt.meta')
saver.restore(sess, tf.train.latest_checkpoint(check_point_path))
graph = tf.get_default_graph()
#print(graph.get_operations())
#with open('op.txt','a') as f:
# f.write(str(graph.get_operations()))
op1 = graph.get_tensor_by_name('fully_connected/biases:0')
print(op1)
使用函數(shù)graph.get_operations()獲取ckpt.meta中保存的graph中的所有operation，而tensor_name為'op_name:0'。
然后使用graph.get_tensor_by_name('op_name:0') 獲取tensor信息。
代碼從ckpt文件中獲取保存的variable的數(shù)據(jù)(tensor的name和value)：
import os
import tensorflow as tf
from tensorflow.python import pywrap_tensorflow
check_point_path = 'variables'
#checkpoint_path = os.path.join(logs_train_dir, 'model.ckpt')
ckpt = tf.train.get_checkpoint_state(checkpoint_dir=check_point_path)
checkpoint_path = os.path.join('.', ckpt.model_checkpoint_path)
#print(ckpt.model_checkpoint_path)
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print("tensor_name: ", key)
#print(reader.get_tensor(key))
法二：
from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file
print_tensors_in_checkpoint_file("variables/save_variables.ckpt",tensor_name='', all_tensors=False, all_tensor_names=False)
注意：tf.train.latest_checkpoint(check_point_path) 方法用來獲取最后一次ckeckpoint的路徑，等價于
ckpt = tf.train.get_checkpoint_state(check_point_path) ckpt.model_checkpoint_path
不能將tf.train.latest_checkpoint與tf.train.get_checkpoint_state 搞混了
以上這篇tensorflow 實現(xiàn)從checkpoint中獲取graph信息就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python中的正斜杠與反斜杠實例驗證,本文給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-03-03
這篇文章主要介紹了Python中super關(guān)鍵字用法,實例分析了super關(guān)鍵字的功能及相關(guān)使用技巧,需要的朋友可以參考下2015-05-05
Linux系統(tǒng)一般自帶Python，有時候又自己下載了Python，因此有可能Python2和Python3同時存在。那么當我們在Terminal鍵入python的時候，會調(diào)出哪個Python呢？本文即解決這個問題2018-10-10
Python 16進制與中文相互轉(zhuǎn)換的實現(xiàn)方法
今天小編就為大家分享一篇Python 16進制與中文相互轉(zhuǎn)換的實現(xiàn)方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-07-07
這篇文章主要介紹了python mysql中in參數(shù)化說明，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
python實現(xiàn)mysql的單引號字符串過濾方法
這篇文章主要介紹了python實現(xiàn)mysql的單引號字符串過濾方法,以一個較為詳細的實例形式分析了Python針對MySQL的操作及字符串過濾的技巧,具有一定參考借鑒價值,需要的朋友可以參考下2015-11-11
這篇文章主要介紹了詳細解析Python中的變量的數(shù)據(jù)類型,是Python學(xué)習(xí)當中的基礎(chǔ)知識,需要的朋友可以參考下2015-05-05
這篇文章主要介紹了python處理html轉(zhuǎn)義字符的方法,結(jié)合實例形式較為詳細的分析了Python針對常見HTML轉(zhuǎn)義字符處理技巧,具有一定參考借鑒價值,需要的朋友可以參考下2016-07-07
Python連接數(shù)據(jù)庫進行數(shù)據(jù)查詢的操作代碼
這篇文章主要介紹了Python連接數(shù)據(jù)庫進行數(shù)據(jù)查詢的操作代碼,本文給大家介紹的非常詳細,感興趣的朋友跟隨小編一起看看吧2024-06-06
簡單聊聊PyTorch里面的torch.nn.Parameter()
torch.nn.parameter是一個被用作神經(jīng)網(wǎng)絡(luò)模塊參數(shù)的tensor,這是一種tensor的子類,下面這篇文章主要給大家介紹了關(guān)于PyTorch里面的torch.nn.Parameter()的相關(guān)資料,需要的朋友可以參考下2022-02-02