参考: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
运行后生成如下图片(截取部分)