caffe可视化绘制网络结构图方法及遇到问题总结

1、准备

       首先要先配置python接口。详细方法我之前的博客中有，配置方法链接：点击打开链接

2、绘制网络结构图

       首先我们要知道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安装