Caffe学习系列(1): 绘制网络模型

1、安装必要的库。

$ sudo apt-get install GraphViz

$ sudo pip install pydot

2、绘制网络

参数说明：

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

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

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

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

例子：

$sudo python python/draw_net.py  examples/mnist/lenet_train_test.prototxt net.png --rankdir=BT

3、绘制自己的网络模型

未完待续。。。