Win10环境下用draw_net.py绘制Caffe模型的网络结构图

1.编译pycaffe项目

Caffe提供了Python接口，用Visual Studio 2013打开caffe-master\window目录下的Caffe.sln，修改文件CommonSettings.props的PythonSupport项为true，并在PythonDir项添加本机的Python安装路径，推荐用Python2，Python3编译时会报错，例如：

...
<PythonSupport>true</PythonSupport>
...
<PythonDir>C:\python27</PythonDir>
...

然后生成pycaffe项目。

2.为Python添加Caffe模块

上一步成功编译pycaffe后，在caffe-master\Build\x64\Release目录下可以找到pycaffe文件夹，把里面的caffe文件夹拷贝到Python安装目录下的Lib\site-packages中，然后用CMD打开Python并尝试导入caffe模块：

import caffe

#如果正常导入，说明添加成功，否则需要根据错误提示安装所需依赖包。
#所需依赖至少包括numpy、scikit-image、scipy、protobuf

3.安装绘图脚本所需依赖

1.安装pydot包
直接打开CMD输入以下命令用pip安装：

python -m pip install pydot

2.安装Graphviz软件:

1. 在Graphviz官网下载安装包；

2. 安装Graphviz并将安装目录下的\bin添加到系统环境变量Path；

3. 以管理员方式打开CMD运行以下命令完成配置:

   dot -c
   

4.用CMD调用Python执行绘图脚本draw_net.py

draw_net.py位于caffe-master\python目录，可以把自己的网络模型配置文件train_test.prototxt拷贝到该目录，并在该目录运行CMD，然后执行以下命令：

python draw_net.py --rankdir LR train_test.prototxt 网络结构图.jpg

#参数--rankdir LR 用来设置网络结构图的摆放方向，可选项有：
#自左向右：LR
#自右向左：RL
#自下而上：BT
#自上而下：TB

就会在caffe-master\python目录下生成“网络结构图.jpg”。