caffe用draw_net.py绘制结构图

step1

在caffe安装目录找到draw_net.py，将其复制到训练模型路径下

step2

安装python-pydot和graphviz

这里我用的是anaconda，所以用conda安装(注意，用conda安装python-pydot是找不到这个包的，这里我直接conda install pydot)

安装graphviz

终端命令   ：[email protected]:~/face_detect$ conda install graphviz

安装python-pydot

终端命令   ：[email protected]:~/face_detect$ conda install pydot 

step3

接下来就可以直接绘图了

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

绘制垂直排列的结构图

终端命令：[email protected]:~/face_detect$ python draw_net.py --rankdir TB train.prototxt train1.png

绘制水平排列的结构图

终端命令：python draw_net.py --rankdir LR train.prototxt train.png

参考：https://blog.csdn.net/langb2014/article/details/50970520

              https://blog.csdn.net/dcxhun3/article/details/52237480