從Tensorflow模型文件中解析並顯示網絡結構圖(pb模型篇)

Tensorflow官方提供的Tensorboard可以可視化神經網絡結構圖,但是說實話,我幾乎從來不用。主要是因爲Tensorboard中查看到的圖結構太混亂了,包含了網絡中所有的計算節點(讀取數據節點、網絡節點、loss計算節點等等)。更可怕的是,如果一個計算節點是由多個基礎計算(如加減乘除等)構成,那麼在Tensorboard中會將基礎計算節點顯示而不是作爲一個整體顯示(典型的如Squeez
相關文章
相關標籤/搜索