NN-SVG: 一個用於生成出版級神經網路架構圖的參數化生成器

NN-SVG: 一個用於生成出版級神經網路架構圖的參數化生成器

它解決了什麼問題

NN-SVG 解決了在學術論文和網頁中手動繪製神經網路架構圖這一既乏味又耗時的過程。研究人員不再需要從頭開始手動構建這些圖表,而是可以透過參數化方式來生成它們。

運作原理

該工具使用 JavaScript 函式庫,根據使用者定義的大小、顏色和佈局參數,以程式化方式生成圖表。它支援三種特定的架構繪圖風格:

  • 全連接神經網路 (FCNN): 使用 D3.js 函式庫生成。
  • 卷積神經網路 (CNN): 基於 LeNet 風格,使用 D3.js 生成。
  • 卷積神經網路 (Deep): 基於 AlexNet 風格,使用 Three.js 生成類似 3D 的表示方式。

對象是誰

需要模型架構出版級圖表的機器學習研究人員,以及可能將此工具用於教學目的的教育工作者。

重點功能

  • 參數化生成: 無需手動繪製即可建立圖表。
  • SVG 匯出: 將圖表匯出為可縮放向量圖形 (SVG),以便高品質地納入論文中。
  • 多種風格: 支援 FCNN、LeNet 風格的 CNN 以及 AlexNet 風格的深度網路。
  • 可自定義樣式: 可控制大小、顏色和佈局參數。

Sources