Caffe小玩意(1)-可视化网络结构

Caffe小玩意(1)-可视化网络结构

$\quad$最近在学习Caffe，但是作为曾经的Windows深度用户，还是比较习惯可视化的界面。然而，Caffe当然是在Linux/OS X系统下更好啦，因为一般还是写script在命令行里面玩的。所以这样就不直观咯，为了能直观地看清楚网络结构，而不是看prototxt脑补…可视化就很重要了。
$\quad$幸好，开发Caffe的大神们已经考虑过这个问题了。在Caffe的根目录下，有个python文件夹，里面有个draw_net.py，就是我们所需要的文件了。
$\quad$接下来我随便挑了一个网络结构。假设我们现在已经在这个python文件夹的路径下，在命令行里，输入

python draw_net.py ../models/bvlc_reference_caffenet/train_val.prototxt vis.png

$\quad$我用的是Macbook pro，在第一次运行的时候报错了

pydot.InvocationException: GraphViz’s executables not found

查了一下发现我缺少了Graphviz这个专门用于画图的软件，所以必须得装一下。如果你装了homebrew，那就简单咯

brew install graphviz

$\quad$再次运行，就输出了vis.png到当前路径。打开看看:

$\quad$比较遗憾的是，对于自己定义的layer，目前貌似没有办法进行可视化（我尝试了一下可视化Faster-RCNN结果失败了）。之后我会再看看怎样把自己定义的layer也可视化。

---