打印tensorflow恢復模型中所有變量與操作節(jié)點方式
我就廢話不多說了，大家還是直接看代碼吧！
#參數(shù)恢復
self.sess=tf.Session()
saver = tf.train.import_meta_graph(os.path.join(model_fullpath,'model.ckpt-7.meta'))
module_file = tf.train.latest_checkpoint(model_fullpath)
saver.restore(self.sess, module_file)
variable_names = [v.name for v in tf.trainable_variables()]
variable_names = [v.name for v in tf.global_variables()]
values = self.sess.run(variable_names)
i=0
for k, v in zip(variable_names, values):
i+=1
if k.find('encode')!=-1:
print(f"第 {i} 個variable")
print("Variable: ", k)
print("Shape: ", v.shape)
print(v)
graph = tf.get_default_graph()
all_ops = graph.get_operations()
for el in all_ops:
print(el.name)
輸出結果：
補充知識：TensorFlow：.ckpt文件與.ckpt.meta和.ckpt.index以及.pb文件之間的關系是什么？
再使用 tf.train.Saver() 保存參數(shù)通常會生成以下文件
那么這些文件具體是什么呢？
.ckpt文件：是舊版本的輸出saver.save(sess)，相當于你的.ckpt-data
“checkpoint”：文件僅用于告知某些TF函數(shù)，這是最新的檢查點文件。
.ckpt-meta：包含元圖，即計算圖的結構，沒有變量的值（基本上你可以在tensorboard / graph中看到）。
.ckpt-data：包含所有變量的值，沒有結構。
.ckpt-index：可能是內部需要的某種索引來正確映射前兩個文件，它通常不是必需的
你可以只用 .ckpt-meta 和恢復一個模型 .ckpt-data
要在python中恢復模型，您通常會使用元數(shù)據(jù)和數(shù)據(jù)文件（但您也可以使用該.pb文件）：
saver = tf.train.import_meta_graph(path_to_ckpt_meta) saver.restore(sess, path_to_ckpt_data)
該.pb文件可以保存您的整個圖表（元+數(shù)據(jù)）
要在c ++中加載和使用（但不訓練）圖形，您通常會使用它來創(chuàng)建freeze_graph，它會.pb從元數(shù)據(jù)和數(shù)據(jù)創(chuàng)建文件。
要小心，（至少在以前的TF版本和某些人中）py提供的功能freeze_graph不能正常工作，所以你必須使用腳本版本。
以上這篇打印tensorflow恢復模型中所有變量與操作節(jié)點方式就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
python?針對在子文件夾中的md文檔實現(xiàn)批量md轉word
這篇文章主要介紹了python?針對在子文件夾中的md文檔實現(xiàn)批量md轉word，但是自己保存的md文檔在不同的文件夾，而大部分只能實現(xiàn)同一文件夾內的轉換，得出下列總結,需要的朋友可以參考一下2022-04-04
python針對mysql數(shù)據(jù)庫的連接、查詢、更新、刪除操作示例
這篇文章主要介紹了python針對mysql數(shù)據(jù)庫的連接、查詢、更新、刪除操作,結合實例形式詳細分析了Python操作mysql數(shù)據(jù)庫的連接與增刪改查相關實現(xiàn)技巧,需要的朋友可以參考下2019-09-09
在本篇文章里小編給大家整理的是一篇關于scrapy處理python爬蟲調度的相關內容，有興趣的朋友們學習下。2020-11-11
這篇文章主要介紹了Python流程控制語句的的相關資料，幫助大家更好的理解和學習python，感興趣的朋友可以了解下2020-10-10
這篇文章主要介紹了python裝飾器的特性原理詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-12-12
這篇文章主要介紹了Pyinstaller 打包發(fā)布經驗總結,使用Pyinstaller打包Python項目包含了大量的坑，感興趣的可以一起來了解一下2020-06-06
Python如何使用標準庫tmpfile庫創(chuàng)建臨時文件
這篇文章主要介紹了Python如何使用標準庫tmpfile庫創(chuàng)建臨時文件問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-02-02
教你使用conda虛擬環(huán)境管理(創(chuàng)建、激活、重命名、刪除虛擬環(huán)境)
conda是一個強大的Python包管理和環(huán)境管理工具,它可以幫助我們輕松地安裝、更新、卸載和切換不同版本的Python和各種第三方庫,本文就來介紹一下conda虛擬環(huán)境管理(創(chuàng)建、激活、重命名、刪除虛擬環(huán)境),感興趣的可以了解一下2024-01-01
在本篇文章里小編給大家整理了關于下載官網(wǎng)python并安裝的步驟詳解，需要的朋友們參考學習下。2019-10-10
Python繼承和子類從Person到Student演示實例
這篇文章主要為大家介紹了Python繼承和子類從Person到Student演示實例分析，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2023-10-10