Tensorflow之pb文件分析

[原文連接](https://blog.csdn.net/u014061630/article/details/80557028

使用Tensorboard分析pb文件有兩種方法:

方法一:

  • 利用pb文件恢復(fù)計算圖

  • 利用Tensorboard查看計算圖的結(jié)構(gòu)

方法二

方法一

1.從pb文件中恢復(fù)計算圖

import tensorflow as tf

model = 'model.pb') #請將這里的pb文件路徑改為自己的
graph = tf.get_default_graph()
graph_def = graph.as_graph_def()
graph_def.ParseFromString(tf.gfile.FastGFile(model, 'rb').read())
tf.import_graph_def(graph_def, name='graph')
summaryWriter = tf.summary.FileWriter('log/', graph)

2.利用Tensorboard查看計算圖

在命令行運行以下命令谴垫,啟動Tensorboard

#命令行運行里執(zhí)行
tensorboard --logdir log/ #這里的路徑就是1中最后一行圖保存的路徑,請根據(jù)自己的需要更改

方法二

利用tools里面的import_pb_to_tensorboard.py工具

#命令行
python -m tensorflow.python.tools.import_pb_to_tensorboard \
  --model_dir="your_path/model.pb" 
  --log_dir="your_log_path"  
tensorboard --logdir="your_log_path" #啟動tensorboard  

或者

#python3
from tensorflow.python.tools.import_pb_to_tensorboard import import_to_tensorboard
model = os.path.join(model_dir, 'tensorflow_inception_graph.pb')
import_to_tensorboard(model_dir=model, log_dir='log/') 
#命令行
tensorboard --logdir="your_log_path" #啟動tensorboard

經(jīng)過查看源碼,第二種方法其實是對第一種方法的包裝恕出。
兩種方法是一致的,只不過第二種方法更加便捷的畴。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末班巩,一起剝皮案震驚了整個濱河市逊桦,隨后出現(xiàn)的幾起案子睡陪,更是在濱河造成了極大的恐慌兰迫,老刑警劉巖,帶你破解...
    沈念sama閱讀 206,214評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件跷车,死亡現(xiàn)場離奇詭異善玫,居然都是意外死亡,警方通過查閱死者的電腦和手機只洒,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,307評論 2 382
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人银舱,你說我怎么就攤上這事〕锨罚” “怎么了?”我有些...
    開封第一講書人閱讀 152,543評論 0 341
  • 文/不壞的土叔 我叫張陵漾岳,是天一觀的道長轰绵。 經(jīng)常有香客問我,道長尼荆,這世上最難降的妖魔是什么左腔? 我笑而不...
    開封第一講書人閱讀 55,221評論 1 279
  • 正文 為了忘掉前任,我火速辦了婚禮捅儒,結(jié)果婚禮上翔悠,老公的妹妹穿的比我還像新娘。我一直安慰自己野芒,他們只是感情好蓄愁,可當(dāng)我...
    茶點故事閱讀 64,224評論 5 371
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著狞悲,像睡著了一般撮抓。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上摇锋,一...
    開封第一講書人閱讀 49,007評論 1 284
  • 那天丹拯,我揣著相機與錄音,去河邊找鬼荸恕。 笑死乖酬,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的融求。 我是一名探鬼主播咬像,決...
    沈念sama閱讀 38,313評論 3 399
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼生宛!你這毒婦竟也來了县昂?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 36,956評論 0 259
  • 序言:老撾萬榮一對情侶失蹤陷舅,失蹤者是張志新(化名)和其女友劉穎倒彰,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體莱睁,經(jīng)...
    沈念sama閱讀 43,441評論 1 300
  • 正文 獨居荒郊野嶺守林人離奇死亡待讳,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 35,925評論 2 323
  • 正文 我和宋清朗相戀三年芒澜,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片创淡。...
    茶點故事閱讀 38,018評論 1 333
  • 序言:一個原本活蹦亂跳的男人離奇死亡痴晦,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出辩昆,到底是詐尸還是另有隱情阅酪,我是刑警寧澤,帶...
    沈念sama閱讀 33,685評論 4 322
  • 正文 年R本政府宣布汁针,位于F島的核電站术辐,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏施无。R本人自食惡果不足惜辉词,卻給世界環(huán)境...
    茶點故事閱讀 39,234評論 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望猾骡。 院中可真熱鬧瑞躺,春花似錦、人聲如沸兴想。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,240評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽嫂便。三九已至捞镰,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間毙替,已是汗流浹背岸售。 一陣腳步聲響...
    開封第一講書人閱讀 31,464評論 1 261
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留厂画,地道東北人凸丸。 一個月前我還...
    沈念sama閱讀 45,467評論 2 352
  • 正文 我出身青樓,卻偏偏與公主長得像袱院,于是被迫代替她去往敵國和親屎慢。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 42,762評論 2 345

推薦閱讀更多精彩內(nèi)容