caffe系列（如何画网络结构图）

caffe系列（如何画网络结构图）

1、安装ＧraphViz

• mac ：brew install graphviz
• linux：sudo apt-get install GraphViz

2、安装 pydot

pip install pydot

3、绘制Lenet模型

sudo python /Users/douglaswang/github/caffe-master/python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT

第一个参数：网络模型的prototxt文件
第二个参数：保存的图片路径及名字
第二个参数：--rankdir=x , x 有四种选项，分别是LR, RL, TB, BT 。用来表示网络的方向，分别是从左到右，从右到左，从上到小，从下到上。默认为ＬＲ。