Caffe學習系列(1): 繪製網絡模型

1、安裝必要的庫。

$ sudo apt-get install GraphViz
$ sudo pip install pydot

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

$sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt net.png --rankdir=BT


3、繪製自己的網絡模型
未完待續。。。