python绘制caffe实例的网络模型

在所有目录与文件均是基于caffe根目录！！！！！

python/draw_net.py可以将网络模型由prototxt变成一张图片。

在绘制之前，先安装两个库：：：

１、安装ＧraphViz

sudo apt-get install GraphViz

Graphviz的是一款图形绘制工具，用来被python程序调用绘制图片。

2、安装pydot

sudo pip install pydot

pydot是python的支持画图的库

编译pycaffe

在绘制之前，还要执行下述语句编译_caffe.cpp

make pycaffe

否则会报如下图所示错误：：：



完成上述三步后，就可以调用脚本来绘制图片了

draw_net.py执行的时候带三个参数

第一个参数：网络模型的prototxt文件

第二个参数：保存的图片路径及名字

第二个参数：–rankdir=x , x 有四种选项，分别是LR, RL, TB, BT 。用来表示网络的方向，分别是从左到右，从右到左，从上到小，从下到上。默认为ＬＲ。

绘制Lenet模型

python python/draw_net.py examples/mnist/lenet_train_test.prototxt ./sc.jpg --rankdir=BT

绘制cifar10的模型

python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt cifar10_net.jpg --rankdir=BT

感谢：：：：

http://www.cnblogs.com/denny402/p/5106764.html

http://blog.csdn.net/keyanxiaocaicai/article/details/51097227