最近看到一個巨牛的人工智能教程，分享一下給大家。教程不僅是零基礎，通俗易懂，而且非常風趣幽默，像看小說一樣！覺得太牛了，所以分享給大家。平時碎片時間可以當小說看，【點這里可以去膜拜一下大神的“小說”】。

Tensorflow官方提供的Tensorboard可以可視化神經網絡結構圖，但是說實話，我幾乎從來不用。主要是因為Tensorboard中查看到的圖結構太混亂了，包含了網絡中所有的計算節點(讀取數據節點、網絡節點、loss計算節點等等)。更可怕的是，如果一個計算節點是由多個基礎計算(如加減乘除等)構成，那么在Tensorboard中會將基礎計算節點顯示而不是作為一個整體顯示(典型的如Squeeze計算節點)。最近為了排查網絡結構BUG花費一周時間，因此，狠下心來決定自己寫一個工具，將Tensorflow中的圖以最簡單的方式顯示最關鍵的網絡結構。

1 Tensor對象與Operation對象

Tensorflow中，Tensor對象主要用于存儲數據如常量和變量(訓練參數)，Operation對象是計算節點，如卷積計算、反卷積計算、ReLU等等。每一個Operation對象均有輸入和輸出Tensor，同理，每個Tensor對象均有對應生成該Tensor的Operation對象和使用該Tensor對象作為輸入的Operation對象。Tensor和Operation對象內均有相關屬性和函數來獲取其關聯的Operation和Tensor對象，相關屬性如下所示。

Tensor對象的op屬性指向生成該Tensor的Operation對象。

Tensor對象的consumers()函數獲取使用該Tensor對象作為輸入的Operation對象。

Operation對象的inputs屬性指向該計算節點的輸入Tensor對象。

Operation對象的outputs屬性執行該計算節點的輸出Tensor對象。

如下圖所示的網絡結構中，調用Tensor_2對象的consumers()函數，返回的是[op_1,op_2]。Tensor_3的op屬性指向的是op_1。op_1的inputs屬性指向的是[Tensor_1,Tensor_2]，op_1的output屬性指向的是[Tensor_3]。

Tensor與Operation

有了Tensor與Operation對應在圖中的關聯關系，就可以將網絡結構給畫出來。

2 提取pb文件中的網絡結構圖

pb文件是將模型參數固化到圖文件中，并合并了一些基礎計算和刪除了反向傳播相關計算得到的protobuf協議文件。如果讀者還不懂如何將CKPT模型文件轉pb文件，請參考我另一篇文章《 Tensorflow MobileNet移植到Android》的第1節部分。有了pb模型文件后，接下來是加載模型，加載pb模型示例代碼如下所示。

def read_graph_from_pb(tf_model_path ,input_names,output_name):

with open(tf_model_path, 'rb') as f:

serialized = f.read()

tf.reset_default_graph()

gdef = tf.GraphDef()

gdef.ParseFromString(serialized)

with tf.Graph().as_default() as g:

tf.import_graph_def(gdef, name='')

with tf.Session(graph=g) as sess:

OPS=get_ops_from_pb(g,input_names,output_name)

return OPS

其中，倒數第2行調用到的函數get_ops_from_pb()用于獲取網絡結構圖中指定輸入節點和指定輸出節點之間的計算節點。之所以要指定輸入和輸出，是為了將輸入之前的計算節點(如加載數據隊列等相關計算節點)和輸出之后的計算節點(如計算loss等相關計算節點)去除，免得礙眼。函數get_ops_from_pb()實現代碼如下。

def get_ops_from_pb(graph,input_names,output_name,save_ori_network=True):

if save_ori_network:

with open('ori_network.txt','w+') as w:

OPS=graph.get_operations()

for op in OPS:

txt = str([v.name for v in op.inputs])+'---->'+op.type+'--->'+str([v.name for v in op.outputs])

w.write(txt+'\n')

inputs_tf = [graph.get_tensor_by_name(input_name) for input_name in input_names]

output_tf =graph.get_tensor_by_name(output_name)

OPS =get_ops_from_inputs_outputs(graph, inputs_tf,[output_tf] )

with open('network.txt','w+') as w:

for op in OPS:

txt = str([v.name for v in op.inputs])+'---->'+op.type+'--->'+str([v.name for v in op.outputs])

w.write(txt+'\n')

OPS = sort_ops(OPS)

OPS = merge_layers(OPS)

return OPS

在裁剪網絡結構(即只保留input_names和output_name之間節點)之前，先將原始的網絡結構寫入到ori_network.txt中，文件中，每一行寫入：輸入Tensor---->op---->輸出Tensor。接下來調用函數get_ops_from_inputs_outputs獲取指定節點之間的節點。并調用sort_ops函數對所有的節點排序，以保證被依賴的節點總是出現在相關節點之前。最后調用merge_layers函數，將一些可以合并的計算合并成一個獨立的節點，例如，Squeeze計算相關節點合并成一個單獨的Squeeze節點，又如const-->identity兩個計算節點可以直接忽略(即刪除)。

注意：篇幅有限，這里不再將函數get_ops_from_inputs_outputs、sort_ops、merge_layers貼出，相關代碼請前往文尾提供的源碼地址中閱讀。

3 繪制網絡結構

考慮到SVG繪制圖形的簡單易用優點，將排好序的網絡計算節點和相關Tensor對象數據以Javascript字符串的形式寫入到HTML中，使用標簽繪制箭頭，使用標簽繪制矩形，使用標簽繪制橢圓，使用標簽顯示文字。繪制類似于如下所示圖像

繪制網絡結構示例

注意：篇幅有限，這里不再介紹Javascript代碼解析模型結構和SVG顯示相關的原理，相關代碼請前往文尾提供的源碼地址中閱讀。

4 測試模型顯示

以《MobileNet V1官方預訓練模型的使用》文中介紹的MobileNet V1網絡結構為例，下載MobileNet_v1_1.0_192文件并壓縮后，得到mobilenet_v1_1.0_192_frozen.pb文件。我們還需要知道mobilenet_v1_1.0_192_frozen.pb模型對應的輸入和輸出Tensor對象的名稱，好在MobileNet_v1_1.0_192壓縮包中包含文件mobilenet_v1_1.0_192_info.txt。通過該文件可知，輸入Tensor的名稱為：input:0，輸出Tensor名稱為：MobilenetV1/Predictions/Reshape_1:0。有了這些信息后，調用函數read_graph_from_pb得到靜態圖的節點列表對象ops，調用函數gen_graph(ops,"save/path/graph.html")后，在目錄save/path中得到graph.html文件，打開graph.html后，顯示結果如下。

顯示網絡結構分兩種模式：合并模式和展開模式，分別如下圖所示。

合并模式網絡結構

截取的展開模式網絡結構

5 源碼地址

總結

以上是生活随笔為你收集整理的tensorflow打印模型图_从Tensorflow模型文件中解析并显示网络结构图（pb模型篇）...的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。