【caffe】繪制網絡結構圖

@tags caffe 網絡結構 可視化

當拿到一份網絡定義文件net.prototxt，可以用工具畫出網絡結構。最快速的方法是使用在線工具netscope，粘貼內容后shift+回車就可以看結果了。

caffe也自帶了網絡結構繪制工具，需要稍微配置下，並確保你用的caffe版本中實現了網絡中涉及到的層。以下是在windows下嘗試並配置的步驟。

在<caffe_root>/python目錄下，看起來draw_net.py腳本提供了繪制網絡結構的功能。不過不能直接執行它，需要在用visual studio編譯caffe項目的時候，配置python支持（推薦裝anaconda2並且配置阿里雲的pip源），然后：

1 環境變量

創建系統環境變量PythonPath，內容為：

<caffe_root>\Build\x64\Release\pycaffe

<caffe_root>換成你自己的目錄，比如我的是d:\lib\caffe-master

2 pip包

用pip安裝一些必要的包

pip install protobuf pydot

3 安裝graphviz

http://www.graphviz.org/Download_windows.php 到這里下載。安裝后把 /bin目錄追加到系統環境變量PATH上

4 繪圖

重開一個cmd控制台，根據prototxt來繪制網絡圖片，

draw_net.py執行的時候帶三個參數

第一個參數：網絡模型的prototxt文件

第二個參數：保存的圖片路徑及名字

第二個參數：--rankdir=x , x 有四種選項，分別是LR, RL, TB, BT 。用來表示網絡的方向，分別是從左到右，從右到左，從上到小，從下到上。默認為ＬＲ。

d:
cd d:\lib\caffe-master\Build\x64\Release\pycaffe
# python draw_net.py prototxt文件名 保存圖片文件名 --rankdir=方向，比如：
python draw_net.py d:\lib\caffe-master\examples\mnist\lenet_train_test.prototxt d:\lib\caffe-master\examples\mnist\lenet_train_test.png --rankdir=LR

看看lenet_train_test.prototxt網絡結構繪制出來是什么樣子的：