所有目录与文件均是基于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。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。
执行:python python/draw_net.py models/bvlc_reference_caffenet/train_val.prototxt caffenet.png
(caffe根目录下,在terminal里运行,不是进入Python再运行)生成的图片在caffe目录下,也可以自己指定目录