1、Netscope–支持Caffe的神经网络结构在线可视化工具
Netscope是个支持prototxt格式描述的神经网络结构的在线可视工具,支持从GitHub Gist或者编辑器中可视化Caffe的网络结构。网址:http://ethereon.github.io/netscope/quickstart.html
使用方法:
(1)打开网页http://ethereon.github.io/netscope/#/editor
(2)将描述神经网络结构的prototxt文件
拖到编辑框里,按shift+enter
2、使用Caffe根目录下python/draw_net.py脚本绘制网络模型
(1)安装环境
在绘制前,需要先安装两个库:GraphViz和pydot
。Graphviz是一个开源的可视化软件,能够以抽象的图和网络表示结构信息,广泛使用于网络、生物信息学、软件工程等领域。
# 安装GraphViz
sudo apt-get install graphviz
# 安装pydot
sudo pip install pydot
(2)开始绘制
draw_net.py
执行的时候需要指定三个参数:
1)网络模型prototxt文件;
2)绘制的图片名字及保存路径,支持png、jpg,不支持保存bmp格式图片;
3)–rankdir=x,x有四种选项:LR(默认)、RL、TB、BT。用来指定网络的绘制方向,分别为从左到右(LR),从右到左(RL),从上到小(TB),从下到上(BT)。
3、总结
Netscope绘制出的图直观简洁,易于快速了解网络模型,但是缺少层内的细节信息;
draw_net.py绘制出的结构图保存了参数信息,细节更丰富,但是结构不是很清晰明了,这一点在大型模型上的体现尤为明显。