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

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

目錄：

一，安裝依賴庫。

二，draw_net.py使用說明。

 

正文：

一，安裝依賴庫。

在繪製以前，須要先安裝相關的依賴庫
１、安裝ＧraphViz
# 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 。用來表示網絡的方向，分別是從左到右，從右到左，從上到小，從下到上。默認爲ＬＲ。

例：繪製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在線可視化，本人親自玩了幾回，發現網址很差用了。