交互改变参数、360度旋转,这个工具让你不用从头构建NN架构图

人工智能 新闻
目前,NN-SVG 工具支持了全连接神经网络(FCNN)、一类卷积神经网络(CNN)和深度神经网络(DNN)的图构建。

神经网络(NN)架构图制作起来往往费时耗力,很多时候机器学习研究人员需要从头开始构建相关图。

近日,机器之心在 GitHub 上发现这样一个项目 ——NN-SVG,可以为研究人员解决这一烦心事。项目作者 Alexander Lenail 为 MIT 计算系统生物学二年级博士生,研究兴趣主要是创建用于生物分析、理解和设计的计算工具。

项目地址:https://github.com/alexlenail/NN-SVG

据作者介绍,NN-SVG 是一个参数化而非手动构建神经网络架构图的交互工具。该工具还能够将构建的图导出至可缩放矢量图形(Scalable Vector Graphics, SVG)文件,适合用在学术论文或网页中。

具体地,NN-SVG 工具可以构建以下三种风格的神经网络图:

  • 全连接神经网络(FCNN)
  • LeNet 论文中提出的一类卷积神经网络(CNN)
  • 风格类似于 AlexNet 论文中提出的深度神经网络(DNN)

其中,前两类神经网络图形使用 JavaScript 库 D3.js 完成的,后一类由 Three.js 完成。NN-SVG 的亮点在于可以根据使用者的喜好来改变神经网络图的大小、颜色和布局参数。

我们仅以 FCNN 风格的生成图为例进行讲解。下图中都是可以改变的组件,如边缘宽度、边缘不透明度、边缘权重对应的颜色、节点直径、层间距、随机权重等。

比如,改变「边缘权重对应的颜色」前后:

再比如,添加「18 和 20 个权重」前后:

LeNet 风格和 AlexNet 风格的神经网络图构建过程与之类似。总之,在这个工具中,一切都是可以交互的。

项目作者希望 NN-SVG 工具可以节省机器学习研究人员的时间,也希望这个软件在某些情境下可以作为教学工具。

感兴趣的小伙伴可以去试试。

NN-SVG 地址:https://alexlenail.me/NN-SVG/LeNet.html

责任编辑:张燕妮 来源: 机器之心Pro
相关推荐

2023-09-19 08:28:32

DiagramsPython工具

2020-09-04 09:55:25

TikTok程序禁令

2023-11-10 08:20:13

2024-04-07 10:21:48

2016-09-28 22:04:01

GNUAutotoolLinux

2020-06-15 08:54:46

架构图 EA业务建模

2020-07-07 07:30:33

技术IT架构

2021-12-09 07:54:18

应用程序架构

2021-04-19 16:15:35

开源开源工具

2011-08-19 11:34:05

iOS架构图

2013-03-20 13:21:51

jQueryjQuery插件

2015-07-09 15:08:08

2021-08-13 07:23:15

架构秒杀系统

2014-08-08 09:03:19

IT性能管理

2024-03-26 09:16:12

网络架构图AWS

2022-04-19 07:51:11

RPC 通信架构

2010-04-21 09:49:26

Windows Pho

2023-09-12 12:11:31

方式计算机交互

2020-04-10 12:12:13

InnoDB存储架构

2021-11-22 10:38:23

架构运维技术
点赞
收藏

51CTO技术栈公众号