caffe 网络模型可视化python例程

所有目录与文件均是基于caffe根目录。

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

（1）安装graphviz（graphviz的是一款图形绘制工具，用来被python程序调用绘制图片）

未安装graphviz会出现：Syntax error"dot" not found in path.

安装graphviz不要用pip install安装，否则还是会找不到可执行程序

执行：sudo apt-get install graphviz

（2）安装pydot

执行：sudo pip install pydot

绘制网络模型图

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

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

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

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

执行：python python/draw_net.py models/bvlc_reference_caffenet/train_val.prototxt caffenet.png

(caffe根目录下,在terminal里运行，不是进入Python再运行)生成的图片在caffe目录下，也可以自己指定目录