可视化caffe的网络模型

1. Netscope

进入http://ethereon.github.io/netscope/#/editor

将prototxt文件复制到左侧编辑框中，然后shift+enter

2. 终端

首先，安装这些东西：

sudo apt-get install graphviz
sudo pip install pydot

然后，将模型的prototxt文件放到draw_net.py（在caffe里有这个函数，找找）同目录下，终端输入：

python draw_net.py --rankdir TB train.prototxt train.png

 

如图（部分）：