caffe用draw_net.py绘制结构图

step1

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

step2

安装python-pydot和graphviz

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

安装graphviz

终端命令   ：fier@fier-PC:~/face_detect$ conda install graphviz

安装python-pydot

终端命令   ：fier@fier-PC:~/face_detect$ conda install pydot 

step3

接下来就可以直接绘图了（竖直绘制用TB，水平绘制用LR）

绘制垂直排列的结构图

终端命令：fier@fier-PC:~/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