caffe绘制网络模型

使用Caffe提供的draw_net.py

1. 安装graphviz :   sudo apt-get install graphviz

2. 安装pydot :  sudo pip install pydot

draw_net.py, 这个文件，就是用来绘制网络模型的。也就是将网络模型由prototxt变成一张图片。

使用范例：sudo  python X Y Z K

X：调用draw_net.py，注意包含路径

Y：网络模型的prototxt文件，注意包含路径

Z：生成的图片的路径及名字

K：rankdir=x，x有四种选择，分别是LR, RL， TB， BT，用来表示网络的方向，分别是从左到右，从右到左，从上到下，从下到上，默认是LR。

示例：以绘制Lenet模型为例：

sudo python /home/xx/caffe/python/draw_net.py /home/xx/caffe/examples/mnist/mnist_autoencoder.prototxt a.png --rankdir=BT