Caffe——绘制网络结构图

１、安装graphViz

打开网址，下载2.38稳定版本的graphViz
点击下载graphviz-2.38.msi，下载完成后，双击msi文件进行安装，安装完成之后，配置环境变量，我的电脑默认安装在：C:\Program Files (x86)\Graphviz2.38该目录下，将该路径C:\Program Files (x86)\Graphviz2.38\bin配置环境变量。
验证： 进入windows命令行界面，输入dot -version，然后按回车，如果显示graphviz的相关版本信息，则安装配置成功。

2 、安装pydot和pydotplus

pip install pydot
pip install pydotplus

3、代码实现绘制结构图

python D:\Caffe\caffe-windows\python\draw_net.py D:\LSLS\caffe\alexnet_lmdb\alexnet_train.prototxt D:\LSLS\caffe\alexnet_train.png --rankdir=BT

直接在pycharm里运行draw_net.py，配置好参数，运行能出结构图。但是在cmd终端执行上述命令就会报错，最终发现是在终端import caffe出了问题，但是在pycharm里面import caffe没有报错

第一个参数：网络模型的prototxt文件

第二个参数：保存的图片路径及名字

第二个参数：–rankdir=x , x 有四种选项，分别是LR, RL, TB, BT 。用来表示网络的方向，分别是从左到右，从右到左，从上到小，从下到上。默认为ＬＲ。