Caffe（18）——绘制网络模型

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

# sudo apt-get install GraphViz

2 安装pydot

# sudo pip install pydot

安装好后就可以用脚本来绘制图片来
draw_net.py执行的时候带三个参数
1）网络模型prototxt文件
2）保存的图片路径及名字
3）--rankdir=x,x有四种选项，分别是LR，RL，TB，BT。用来表示网络方向，从左到右分，从右到左，从上到下，从下到上。默认为LR。
例：绘制Lenet

# sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT

# sudo python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT