1. 程式人生 > >caffe中網路結構的視覺化

caffe中網路結構的視覺化

實驗工具是 ubuntu系統下的caffe

視覺化方法很多,一種是用caffe自帶的draw_net.py 來實現網路結構的視覺化。具體實現如下

>> python  /caffe/python/draw_net.py  train.prototxt  net.png  
用python命令執行draw_net.py   實現train.prototxt 的網路結構 並儲存在net.png 中,net.png 在caffe/python的資料夾下。這樣畫出來的是從左到右 程式碼後加上-rankdir BT可以實現從下到上,看你想怎麼看吧


還有一種是GitHub上ethereon開發的一個神經網路視覺化的好工具

Netscope

(網址:http://ethereon.github.io/netscope/#/editor),只需要把prototxt檔案內容複製到左邊的框框中,然後同時按住shift+enter即可在右邊生成網路的結構圖。

在netscope的基礎上,GitHub上的dgschwend又開發了視覺化+分析網路引數、中間變數等功能的加強版(網址:https://dgschwend.github.io/netsope/quickstart.html)