Caffe学习系列 绘制网络模型

转载于 http://www.cnblogs.com/denny402/p/5106764.html

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

在绘制之前，需要先安装两个库

１、安装ＧraphViz

# sudo apt-get install GraphViz

注意，这里用的是apt-get来安装，而不是pip.

2 、安装pydot

# sudo pip install pydot

用的是pip来安装，而不是apt-get

安装好了，就可以调用脚本来绘制图片了

draw_net.py执行的时候带三个参数

第一个参数：网络模型的prototxt文件

第二个参数：保存的图片路径及名字

第二个参数：--rankdir=x , x 有四种选项，分别是LR, RL, TB, BT 。用来表示网络的方向，分别是从左到右，从右到左，从上到小，从下到上。默认为ＬＲ。

例：绘制Lenet模型

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

例：绘制cifar10的模型

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