caffe中網路結構的視覺化
阿新 • • 發佈:2019-01-04
實驗工具是 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)