drawnet.py绘制网络结构

脚本在caffe的python目录里，需要安装pydot，graphviz

安装pydot：

$ sudo apt-get install python-pydot 

安装graphviz：

 

$ sudo apt-get install graphviz  

 

生成的命令行：

python draw_net.py + 网络prototxt + 生成图片名

$ python draw_net.py --rankdir TB ../examples/mnist/lenet.prototxt ../examples/mnist/lenet_structure.jpg

–rankdir选项确定了输出图片的摆放位置，如TB (Top-Bottom) 就是网络从上到下拜访，同理RL (Right-Left）就是从右向左，LR (Left-Right) 就是从左向右摆放。默认使用的是LR方式。注意是--。

生成的可以是jpg，也可以是png，但生成的文件名必须是绝对路径，不能直接写成lenet_structure.jpg，那样会识别不了。

 

 

绘制模型会报的常见错误：

http://blog.csdn.net/yiranyhy/article/details/55509304

 

这里有一个在线的绘制：

http://ethereon.github.io/netscope/#/editor