Caffe(7)--神经网络模型结构可视化

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绘制出的结构图保存了参数信息,细节更丰富,但是结构不是很清晰明了,这一点在大型模型上的体现尤为明显。

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券