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: 將圖表匯出為 SVG 格式,以便高品質地放入論文中。
  • Multiple Styles: 支援 FCNN、LeNet 風格的 CNN 以及 AlexNet 風格的深度網路。
  • Customizable Styling: 可控制尺寸、顏色和佈局參數。"},