Caffe學習系列(18): 繪制網絡模型

python/draw_net.py, 這個文件，就是用來繪制網絡模型的。也就是將網絡模型由prototxt變成一張圖片。

在繪制之前，需要先安裝兩個庫

１、安裝ＧraphViz

# sudo apt-get install GraphViz

注意，這里用的是apt-get來安裝，而不是pip.

2 、安裝pydot

# sudo pip install pydot

用的是pip來安裝，而不是apt-get

安裝好了，就可以調用腳本來繪制圖片了

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=BT

例：繪制cifar10的模型

#  sudo python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT