caffe一些小知识点-caffe自带绘制网络模型

使用Caffe提供的draw_net.py

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

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

2. 安装pydot :  sudo pip install pydot

3.执行draw_net.py文件，带有三个参数：网络模型的prototxt文件，保存图片的路径及名字，--rankdir=x，x有四种选择，分别是LR, RL， TB， BT，用来表示网络的方向，分别是从左到右，从右到左，从上到下，从下到上，默认是LR。

以绘制Lenet模型为例：

python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=TB