netron: 卷積神經網路與機器學習模型架構的視覺化檢視器

netron: 卷積神經網路與機器學習模型架構的視覺化檢視器

它解決了什麼問題

Netron 是一個神經網路與機器學習模型的視覺化工具。它允許使用者透過開啟模型檔案來查看模型的架構,這些檔案通常是一組複雜的數學運算與層級結構。它透過提供圖形化表示來簡化模型檢視的過程。

運作原理

Netron 可以讀取各種格式的模型檔案。它支援廣泛的權重與架構檔案,包括 ONNX、TensorFlow Lite、PyTorch (TorchScript, torch.export, ExecuTorch)、TensorFlow、Core ML、OpenVINO、Keras、Caffe、Darknet、Safetensors 以及 NumPy arrays。此外,它也對 MLIR、JAX、GGUF、RKNN、ncnn、MNN、PaddlePaddle 與 scikit-learn 提供實驗性支援。

對象是誰

它是為需要檢查、除錯或理解預訓練模型結構及其數據流向的開發者與研究人員所設計的。

重點特色

  • 幾乎支援所有主要的 AI 模型格式。
  • 提供瀏覽器工具、桌面應用程式 (macOS, Linux, Windows) 以及 Python 套件形式。
  • 能夠直接從瀏覽器版本開啟範例模型以進行快速測試。

Sources