Caffe---自帶工具進行網絡結構(xxx.prototxt)可視化


Caffe---自帶繪圖工具(draw_net.py)繪制網絡結構圖(xxx.prototxt)

目錄:

一,安裝依賴庫。

二,draw_net.py使用說明。

 

正文:

一,安裝依賴庫。

在繪制之前,需要先安裝相關的依賴庫
1、安裝GraphViz
# sudo apt-get install GraphViz
注意,這里用的是apt-get來安裝,而不是pip.

2 、安裝pydot
# sudo pip install pydot
用的是pip來安裝,而不是apt-get
安裝好了,就可以調用腳本來繪制圖片了

二,draw_net.py使用說明。
draw_net.py執行的時候帶三個參數
第一個參數:網絡模型的prototxt文件
第二個參數:保存的圖片路徑及名字
第二個參數:--rankdir=x , x 有四種選項,分別是LR, RL, TB, BT 。用來表示網絡的方向,分別是從左到右,從右到左,從上到小,從下到上。默認為LR。

例:繪制Lenet模型
sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=LR

#cd caffe
#python python/draw_net.py examples/mnist/lenet_train_test.prototxt my_show_NetStructure/mnist_lenet_train_test.png --rankdir=LR

#python python/draw_net.py examples/mnist/mnist_autoencoder.prototxt my_show_NetStructure/mnist_autoencoder.png --rankdir=LR

#python python/draw_net.py examples/cifar10/cifar10_quick_train_test.prototxt my_show_NetStructure/cifar10_quick_train_test.png --rankdir=LR


 

最后補充,在Caffe中可視化prototxt格式網絡結構的方法:目前除了可以使用Caffe提供的【1】draw_net.py,還可以使用【2】Netscope在線可視化。

Netscope是個支持prototxt格式描述的神經網絡結構的在線可視工具,即:http://ethereon.github.io/netscope/quickstart.html

使用方法:首先打開這個地址:http://ethereon.github.io/netscope/#/editor,然后把你的描述神經網絡結構的prototxt文件復制到該編輯框里,最后按shift-enter,這樣就可以直接以圖形方式顯示網絡的結構 。

使用Netscope在線可視化,本人親自玩了幾次,發現網址不好用了。

 

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM