神经网络（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 的亮点在于可以根据使用者的喜好来改变神经网络图的大小、颜色和布局参数。

内容中包含的图片若涉及版权问题，请及时与我们联系删除