Caffe下使用draw_net.py绘制lenet网络结构图

一.安装库：
python/draw_net.py, 这个文件，就是用来绘制网络模型的。也就是将网络模型由prototxt变成一张图片。

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

１、安装ＧraphViz

sudo su

sudo apt-get insall graphviz ＃这里用的是apt-get来安装，而不是pip.

2 、安装pydot

sudo pip install pydot　＃pip安装pydot
pip uninstall pyparsing 
 　

二．使用draw_net.py绘制lenet网络结构图
安装好了，就可以调用脚本来绘制图片了，draw_net.py在/caffe/python路径下，用于可视化Net

cd caffe/python  #进入draw_net.py目录下

python draw_net.py --rankdir TB ../examples/mnist/lenet.prototxt ../examples/mnist/lenet_structure.jpg

＃–rankdir选项确定了输出图片的摆放位置，有四种选项：LR, RL, TB, BT ，分别是从左到右，从右到左，从上到小，从下到上；默认为ＬＲ

运行时出现了如下的错误：

解决方法如下：
参考：http://blog.csdn.net/tracer9/article/details/50455353

sudo su  #进入root

pip install -Iv https://pypi.python.org/packages/source/p/pyparsing/pyparsing-1.5.7.tar.gz#md5=9be0fcdcc595199c646ab317c1d9a709 

pip install pydot　

python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png　＃重新绘制