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选项确定了