caffe可视化绘制网络结构图

首先我们要知道caffe自带了根据模型prototxt文件作图的python文件，位于$caffe/python文件夹下的draw_net,py文件。执行网络结构可视化之前调用该文件就可以了。具体命令如下：

$cd caffe/python

$python draw_net.py ../models/bvlc_reference_caffenet/train_val.prototxt caffenet.png #caffenet结构图

$python draw_net.py ../models/bvlc_alexnet/train_val.prototxt alexnet.png #alexnet结构图

以下各种结构图不一一列举，只是提供一种方法。

备注：如果想把结构图保存在特定文件夹下，只需在.png前添加所要保存的文件夹路径。

例如：$sudo python draw_net.py ../models/bvlc_reference_caffenet/train_val.prototxt ../jiegoutu/caffenet.png    就是把文件caffenet.png生成并存放在caffe/jiegoutu文件夹下。

3、遇到的问题

（1）ImportError: No module named pydot

解决：未安装pydot；

回到根目录命令：$sudo pip install pydot

（2）Exception: “dot” not found in path

解决：未安装graphviz

回到根目录命令：$sudo apt-get install graphviz #切记不可使用pip安装