tensorflow 實現(xiàn)打印pb模型的所有節(jié)點
更新時間：2020年01月23日 17:39:19 作者：coocoky
今天小編就為大家分享一篇tensorflow 實現(xiàn)打印pb模型的所有節(jié)點，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
只有pd模型文件, 打印所有節(jié)點
from tensorflow.python.framework import tensor_util
from google.protobuf import text_format
import tensorflow as tf
from tensorflow.python.platform import gfile
from tensorflow.python.framework import tensor_util
GRAPH_PB_PATH = 'models/frozen_person_graph.pb' #path to your .pb file
with tf.Session() as sess:
print("load graph")
with gfile.FastGFile(GRAPH_PB_PATH,'rb') as f:
graph_def = tf.GraphDef()
# Note: one of the following two lines work if required libraries are available
#text_format.Merge(f.read(), graph_def)
graph_def.ParseFromString(f.read())
tf.import_graph_def(graph_def, name='')
for i,n in enumerate(graph_def.node):
print("Name of the node - %s" % n.name)
以上這篇tensorflow 實現(xiàn)打印pb模型的所有節(jié)點就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了python基礎之停用詞過濾詳解,文中有非常詳細的代碼示例,對正在學習python基礎的小伙伴們有很好地幫助,需要的朋友可以參考下2021-04-04
這篇文章主要為大家詳細介紹了python實現(xiàn)遺傳算法，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-03-03
解決Pycharm中恢復被exclude的項目問題(pycharm source root)
今天小編就為大家分享一篇解決Pycharm中恢復被exclude的項目問題(pycharm source root)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
這篇文章主要介紹了python selenium登錄豆瓣網過程解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-08-08
本文給大家詳細介紹的是Python數據類型中的列表（list），非常的簡單實用，有需要的小伙伴可以參考下2016-05-05
這篇文章主要介紹了Python列表解析操作,結合實例形式總結分析了Python列表解析常見的列表推導式、循環(huán)語句、條件列表、嵌套列表等相關操作技巧,需要的朋友可以參考下2020-02-02
這篇文章主要介紹了如何遍歷Python列表的方法示例，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-02-02
本文主要介紹了Selenium多窗口切換解決方案，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2022-07-07
Python cookbook（數據結構與算法）將序列分解為單獨變量的方法
這篇文章主要介紹了Python cookbook（數據結構與算法）將序列分解為單獨變量的方法,結合實例形式分析了Python序列賦值實現(xiàn)的分解成單獨變量功能相關操作技巧,需要的朋友可以參考下2018-02-02
pandas是基于NumPy的一種工具，該工具是為了解決數據分析任務而創(chuàng)建的。Pandas納入了大量庫和一些標準的數據模型，提供了高效操作大型數據集的工具。pandas提供大量快速便捷地處理數據的函數和方法。你很快就會發(fā)現(xiàn)，它是使Python強大而高效的數據分析環(huán)境的重要因素之一2022-01-01