簡介

net視覺化主要是以流程圖的形式把caffe中使用的模型（每個layer以及layer和layer之間的連線）展現出來，這樣更加直觀和具體。caffe的python介面下面已經提供了可視net 的功能，就是’draw_net.py’檔案。

編譯pycaffe

編譯caffe時必須編譯了caffe的python的wrapper，命令如下，這樣才能正常使用caffe下的python介面

make pycaffe

安裝pydot

python的pydot模組是Graphviz的dot語言的python介面，這裡主要使用Graphviz繪圖，當然要安裝pydot
網址：https://pypi.python.org/pypi/pydot

source ENV/bin/activate
pip install pydot

安裝Graphviz

Graphviz是大名鼎鼎的貝爾實驗室的幾位牛人開發的一個畫圖工具。它的理念和一般的“所見即所得”的畫圖工具不一樣，是“所想即所得”。Graphviz提供了dot語言來編寫繪圖指令碼。
網址：http://www.graphviz.org/
可以根據自己的系統下載相應的deb檔案安裝。
下載原始碼根據INSTALL編譯安裝，執行下面命令進行編譯安裝。

cd graphviz_dir
./configure
make 
make check
sudo make insall
make installcheck
 

繪製

繪製mnist的LeNet網路層結構

cd caffe_root/python
python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png

結果