Windows下使用python绘制caffe中.prototxt网络结构数据可视化

准备工具：

1. 已编译好的pycaffe

2. Anaconda（python2.7）

3. graphviz

4. pydot 

1. graphviz安装

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

graphviz for windows 下载地址：  http://www.graphviz.org/Download_windows.php

下载完成之后解压出来是一个release的文件夹，把release下的bin文件夹的路径 C:\ProgramData\graphviz\release\bin 添加到环境变量里，让系统可以找到graphviz的相关文件就可以了

2. pydot安装

pydot模块提供了python对graphviz dot语言的接口，使在python下可以调用graphviz工具绘图。

pydot下载地址： https://pypi.python.org/pypi/pydot

下载之后解压，在解压出来的pydot-1.2.3文件夹下有一个pydot.py的安装文件，在所在文件夹下运行cmd命令，输入python pydot.py install 完成安装。

如果安装完之后还是提示找不到pydot模块的话，可以尝试直接在cmd命令下输入 pip install pydot，anaconda会自动下载安装：

3.  .prtotxt网络结构可视化

caffe中的lenet_train_test_prototxt文件定义了一个LeNet卷积神经网络模型，尝试使用python把该模型可视化。

新建一个test.bat文件，输入以下内容：

python D:\Software\Caffe\caffe-master\Build\x64\Release\pycaffe\draw_net.py
D:\Software\Caffe\caffe-master\examples\mnist\lenet_train_test.prototxt
D:\Software\Caffe\prototxtVisualization\prototxtV.pdf

双击运行test.bat文件，执行完之后在相应文件夹路径下生成prototxtV.pdf文件，分成两部分显示：