神经网络结构可视化绘图方法及工具推荐

释放双眼,带上耳机,听听看~!
介绍几种基于在线网页或软件的神经网络模型结构可视化绘图方法,推荐NN-SVG和Netron两种工具,帮助读者更直观地了解神经网络结构及参数变化情况。

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

  本文介绍几种基于在线网页或软件的神经网络模型结构可视化绘图方法。

  之前向大家介绍了一种基于Python第三方ann_visualizer模块的神经网络结构可视化方法,大家可以直接点击Python绘制神经网络模型的图像进行查看;这一方法可以对Dense隐藏层以及MaxPooling层、Dropout层、Flatten层等其它类型的隐藏层加以绘制,功能非常强大,但是需要用代码执行,且在执行前需要将神经网络的全部结构与输入数据配置好后才可以绘制,稍微有一些繁琐。

  今天,就向大家介绍两种新的神经网络可视化绘图方法,其中,一种是在线绘制,一种是基于软件绘制(其也可以在网页中直接绘制)。

1 NN-SVG

  网址:alexlenail.me/NN-SVG/inde…

  NN-SVG是一个在线神经网络结构绘制网页平台,进入后点点鼠标就可以绘制出精美的神经网络图像,且还可以对整幅图像的方向,以及接点、连接线、箭头等等要素的样式、大小、颜色、权重、间隙大小等属性加以调整,可谓非常方便、非常强大。

神经网络结构可视化绘图方法及工具推荐

  在其最下方,可以对输入层、隐藏层与输出层的数量与神经元个数加以调整。

神经网络结构可视化绘图方法及工具推荐

  配置完成取得满意的图像后,大家可以点击最上方的“Download SVG”进行图像下载。需要注意,下载图像的格式是.svg,需要我们手动转换为常见的图片格式。具体在线格式转换网站网上有很多,这里就不再叙述了~

  但是,NN-SVG仅仅可以显示简单的隐藏层,并不能对隐藏层的类型加以区分绘制。

2 Netron

  网址:github.com/lutzroeder/…

  Netron是一个对神经网络以及各类机器学习、深度学习算法进行可视化的工具,分为软件版与在线版(在线版网页:netron.app/ )。绘制时我们需要先建立并保存自己的模型(并且还是要运行之后的,因为其会显示模型中具体参数的变化情况),随后用其打开即可。Netron支持的模型格式如下所示。

神经网络结构可视化绘图方法及工具推荐

  我们以Windows下其软件版本为例介绍。下载安装包后直接安装。

神经网络结构可视化绘图方法及工具推荐

  安装完毕后打开软件,如下所示。

神经网络结构可视化绘图方法及工具推荐

  随后,打开我们保存的模型。

神经网络结构可视化绘图方法及工具推荐

  即可实现具体模型及其每一个参数变化情况的可视化图像,非常具体、细致。

  个人感觉用Netron进行具体带参数分析的可视化比较方便,如果只是想单纯看一下神经网络的结构的话,用其可能不太方便、不太直观。

本网站的内容主要来自互联网上的各种资源,仅供参考和信息分享之用,不代表本网站拥有相关版权或知识产权。如您认为内容侵犯您的权益,请联系我们,我们将尽快采取行动,包括删除或更正。
AI教程

点云数据集与核心算法

2023-12-10 21:08:14

AI教程

扩散模型:文本生成图像领域的核心方法

2023-12-10 21:23:14

个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索