caffe网络模型可视化方法

参考：https://blog.csdn.net/u012915263/article/details/60579429

方法一：利用Netscope工具

caffe的*.prototxt定义了神经网络的结构，为了直观的看到神经网络的连接工具，使用Netscope工具来在线显示网络结构

链接：https://ethereon.github.io/netscope/#/editor

注意该网需要使用

打开后界面如图所示

将*.prototxt中的内容复制到左边黑色文本框中，然后shift+enter即可在右边显示网络结构，效果如下

如图显示了Pvanet的网络结构

方法二：利用draw_net.py脚本

前提是安装好caffe，然后在文件夹中可找到draw_net.py脚本，然后将要可视化的网络结构文件放到同一文件夹下，终端中运行以下命令即可显示

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

其中，TB代表从top画到bottom，test.prototxt是pvanet的网络结构，存储成名字为test.png的图片

rankdir有有四种选项，分别是LR，RL，TB， BT 。用来表示网络的方向，分别是从左到右，从右到左，从上到小，从下到上。默认为LR

运行后会生成test.png的图片

第一次运行可能会报和dot相关的错，需要安装以下两个包

pip install pydot
sudo apt-get install graphviz

运行后生成如下图片（截取部分）