Linux环境

在caffe/python下有draw_net.py这个文件。就是用来绘制网络模型的。也就是将网络模型由prototxt变成一张图片。
在绘制之前，需要先安装两个库。

１、安装ＧraphViz

graphviz是贝尔实验室开发的一个开源的绘图工具，它可以很方便绘制结构化的图形网络，支持多种格式输出，如各种常见的图片格式（bmp、png等），PDF，SVG等。 graphviz使用dot作为脚本语言，只需要在dot脚本中定义图的顶点和边，以及形状、颜色、字体、填充等样式，graphviz就可以使用合适的布局算法对图形布局，使各顶点均匀分布在画布上，并且尽可能的减少边的交叉。

# sudo apt-get install GraphViz

注意，这里用的是apt-get来安装，而不是pip.

2 、安装pydot

# sudo pip install pydot

用的是pip来安装，而不是apt-get
安装好了，就可以调用脚本来绘制图片了
draw_net.py执行的时候带三个参数

第一个参数：网络模型的prototxt文件
第二个参数：保存的图片路径及名字
第二个参数：–rankdir=x , x 有四种选项，分别是LR, RL, TB, BT 。用来表示网络的方向，分别是从左到右，从右到左，从上到小，从下到上。默认为ＬＲ。

3.示例

例：绘制Lenet模型

# sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT

这里写图片描述

例：绘制cifar10的模型

#  sudo python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT

这里写图片描述

windows环境

编译pycaffe后在Builb\x64\Release\pycaffe 下除了会生成caffe文件夹，还会生成几个.py文件，其中就有draw_net.py，或者在caffe/python下也可以找到这个文件。

1. 安装ＧraphViz

在windows下绘图需要安装ＧraphViz的windows版本。graphviz for windows 下载地址：点我
这里写图片描述
graphviz-2.38.msi是安装文件，graphviz-2.38.zip不需要安装直接解压配置环境即可。两个选择任意一个都可以。我选择的是graphviz-2.38.msi，安装完成后把bin目录配置到环境变量。