Python/draw_net.py, 这个文件，就是用来绘制网络模型的。也就是将网络模型由prototxt变成一张图片。
１、安装ＧraphViz
# sudo apt-get install GraphViz
也可以去官网下载，手动安装
2 、安装pydot
sudo apt-get install sudo apt-get install
或者下边的方法编译。
# sudo pip install pydot
3.编译pycaffe
#sudo make pycaffe
编译之前可能需要配置makefile.config文件，计算机需要安装anaconda，并加入路径到配置文件中去。
# Anaconda Python distribution is quite popular. Include path: # Verify anaconda location, sometimes it‘s in root. ANACONDA_HOME := $(HOME)/anaconda2 PYTHON_INCLUDE := $(ANACONDA_HOME)/include $(ANACONDA_HOME)/include/python2.7 $(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include # Uncomment to use Python 3 (default is Python 2) # PYTHON_LIBRARIES := boost_python3 python3.5m # PYTHON_INCLUDE := /usr/include/python3.5m # /usr/lib/python3.5/dist-packages/numpy/core/include # We need to be able to find libpythonX.X.so or .dylib. PYTHON_LIB := /usr/lib PYTHON_LIB := $(ANACONDA_HOME)/lib
安装好了，就可以调用脚本来绘制图片了
第一个参数：网络模型的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
原文：http://www.cnblogs.com/wuchaodzxx/p/5885186.html