Tensorflow: 從checkpoint文件中讀取tensor方式
更新時間：2020年02月10日 14:49:57 作者：noirblack
今天小編就為大家分享一篇Tensorflow: 從checkpoint文件中讀取tensor方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
在使用pre-train model時候，我們需要restore variables from checkpoint files.
經(jīng)常出現(xiàn)在checkpoint 中找不到”Tensor name not found”.
這時候需要查看一下ckpt中到底有哪些變量
import os
from tensorflow.python import pywrap_tensorflow
checkpoint_path = os.path.join(model_dir, "model.ckpt")
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
print("tensor_name: ", key)
print(reader.get_tensor(key))
可以顯示ckpt中的tensor名字和值，當然也可以用pycharm調(diào)試。
以上這篇Tensorflow: 從checkpoint文件中讀取tensor方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
在用Python進行數(shù)據(jù)處理的時候，經(jīng)常會遇到DataFrame中的某一列本應該是數(shù)值類型，但由于數(shù)據(jù)不規(guī)范導致在字段中夾雜了非數(shù)值類型，本文就介紹了Python如何判斷字符串是否僅包含數(shù)字，感興趣的可以了解一下2022-03-03
Python實現(xiàn)簡單線性插值去馬賽克算法代碼示例
去馬賽克是圖像處理中的一項技術,用于從單色彩濾光片陣列（CFA）圖像恢復全彩圖像,本文介紹了一種基于簡單線性插值的去馬賽克算法,并展示了如何將MATLAB代碼轉換為Python代碼,需要的朋友可以參考下2024-10-10
在本文中小編給大家整理了關于python全棧要學什么以及python全棧學習路線的知識點內(nèi)容，需要的朋友們參考下。2019-06-06
matlab中imadjust函數(shù)的作用及應用舉例
這篇文章主要介紹了matlab中imadjust函數(shù)的作用及應用舉例，本文通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2020-02-02
python使用內(nèi)存zipfile對象在內(nèi)存中打包文件示例
這篇文章主要介紹了python使用內(nèi)存zipfile對象在內(nèi)存中打包文件示例,需要的朋友可以參考下2014-04-04
今天小編就為大家分享一篇Python 文本文件內(nèi)容批量抽取實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12
django框架自定義模板標簽(template tag)操作示例
這篇文章主要介紹了django框架自定義模板標簽(template tag)操作,結合實例形式分析了Django框架自定義模板標簽原理、操作步驟與相關實現(xiàn)技巧,需要的朋友可以參考下2019-06-06
這篇文章主要給大家介紹了關于python更換國內(nèi)鏡像源三種實用方法的相關資料,更換Python鏡像源可以幫助解決使用pip安裝包時速度過慢或無法連接的問題,需要的朋友可以參考下2023-09-09
由于工作需要,客戶需要將多個excel文件合并成一個excel中,下面這篇文章主要給大家介紹了關于使用Python快速進行Excel合并的幾種場景,文中通過代碼介紹的非常詳細,需要的朋友可以參考下2023-10-10
Python語法定義了用于在 Python 編程中創(chuàng)建句子的所有規(guī)則集。如果想更深入地研究 Python 詞法結構，需要了解構成語句的句法元素，即構成 Python 程序的基本單元，涵蓋控制結構，在不同代碼組之間引導程序流的構造，快跟隨小編一起學習一下吧2022-04-04