NN-SVG
神经网络架构的图示通常需要花费大量时间来制作,机器学习研究人员往往不得不从头手工构建这些图表。
NN-SVG 是一个用于参数化创建神经网络(NN)架构图的工具,而不是手动绘制。它还提供了将这些图导出为可缩放矢量图形(SVG)文件的功能,适合包含在学术论文或网页中。
该工具提供了三种类型图形的生成能力:经典的全连接神经网络(FCNN)图形、在 LeNet 论文 中介绍的卷积神经网络(CNN)图形,以及在 AlexNet 论文 中介绍的深度神经网络图形。前两者是使用 D3 javascript 库 实现的,后者则使用了 javascript 库 Three.js。NN-SVG 通过多种尺寸、颜色和布局参数,提供了按用户喜好定制图形样式的能力。
我希望这个工具能为机器学习研究人员节省时间,并且我也希望这个软件在某些情况下能作为一种教学工具。
引用
LeNail, (2019). NN-SVG: Publication-Ready Neural Network Architecture Schematics.
Journal of Open Source Software, 4(33), 747, https://doi.org/10.21105/joss.00747