alexlenail/NN-SVG

Publication-ready NN-architecture schematics.

What it solves

NN-SVG 解决了为学术论文和网页手动绘制神经网络架构图这一既乏味又耗时的过程。研究人员不再需要从头开始手动构建这些图表，而是可以通过参数化方式生成它们。

How it works

该工具使用 JavaScript 库来根据用户定义的尺寸、颜色和布局参数，以编程方式生成图形。它支持三种特定的架构绘图风格：

  • Fully-Connected Neural Networks (FCNN): 使用 D3.js 库生成。
  • Convolutional Neural Networks (CNN): 基于 LeNet 风格，使用 D3.js 生成。
  • Convolutional Neural Networks (Deep): 基于 AlexNet 风格，使用 Three.js 生成具有 3D 效果的表示法。

Who it’s for

需要模型架构的可发表性原理图的研究人员，以及可能将该工具用于教学目的的教育工作者。

Highlights

  • Parametric Generation: 创建图表而无需手动绘制。

  • SVG Export: 将图形导出为 Scalable Vector Graphics，以便高质量地包含在论文中。

  • Multiple Styles: 支持 FCNN、LeNet 风格的 CNN 以及 AlexNet 风格的深度网络。

  • Customizable Styling: 控制尺寸、颜色和布局参数。"},