tensorflow查看ckpt各節(jié)點名稱實例
更新時間：2020年01月21日 10:28:57 作者：深夜蟲鳴
今天小編就為大家分享一篇tensorflow查看ckpt各節(jié)點名稱實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
運行下列腳本，可以打印出模型各個節(jié)點變量的名稱：
from tensorflow.python import pywrap_tensorflow
import os
checkpoint_path=os.path.join('model.ckpt-131805')
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
checkpoint_path為自己的模型路徑
以上這篇tensorflow查看ckpt各節(jié)點名稱實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要和大家分享一個簡單的小程序，只需要10行Python代碼，就能助你整理雜亂無章的文件，感興趣的小伙伴可以跟隨小編一起了解一下2023-02-02
django 連接數(shù)據(jù)庫出現(xiàn)1045錯誤的解決方式
這篇文章主要介紹了django 連接數(shù)據(jù)庫出現(xiàn)1045錯誤的解決方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
這篇文章主要介紹了Django初步使用Celery處理耗時任務和定時任務問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-12-12
最近使用python里的matplotlib庫繪圖,想在代碼結束時顯示圖片看看,下面這篇文章主要給大家介紹了關于Vscode使用matplotlib顯示圖像的相關資料,需要的朋友可以參考下2024-03-03
python構造icmp echo請求和實現(xiàn)網(wǎng)絡探測器功能代碼分享
本文分享了二個python示例，python構造icmp echo請求、實現(xiàn)網(wǎng)絡探測器功能代碼，類似nmap功能2014-01-01
SIFT特征的講解已經(jīng)很多了,本文就借助vlfeat對SIFT特征的提取過程做一個總結。接下來通過本文給大家介紹圖像檢索之基于vlfeat實現(xiàn)SIFT，感興趣的朋友跟隨小編一起看看吧2021-12-12
這篇文章主要為大家介紹了如何使用pip安裝清華大學鏡像源步驟詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-10-10
這篇文章主要介紹了python元組列表字典，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-04-04
Python實現(xiàn)將n個點均勻地分布在球面上的方法
這篇文章主要介紹了Python實現(xiàn)將n個點均勻地分布在球面上的方法,涉及Python繪圖的技巧與相關數(shù)學函數(shù)的調(diào)用,具有一定參考借鑒價值,需要的朋友可以參考下2015-03-03
Python+OpenCV手勢檢測與識別Mediapipe基礎篇
網(wǎng)上搜到了一些關于手勢處理的實驗,我在這兒簡單的實現(xiàn)一下,下面這篇文章主要給大家介紹了關于Python+OpenCV手勢檢測與識別Mediapipe基礎篇的相關資料,需要的朋友可以參考下2022-12-12