lutzroeder/netron
Visualizer for neural network, deep learning and machine learning models
What it solves
Netron 是一個用於神經網路和機器學習模型的視覺化工具。它允許使用者透過開啟模型檔案來查看模型的架構，這些檔案通常是一組複雜的數學運算和層級結構。它透過提供圖形化表示來簡化模型檢查的過程。
How it works
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 提供實驗性支援。
Who it’s for
它是為需要檢查、除錯或理解預訓練模型結構及其數據流向的開發者和研究人員而設計的。
Highlights
- Comprehensive support for almost all major AI model formats.
- Available as a browser-based tool, desktop application (macOS, Linux, Windows), and a Python package.
- Ability to open sample models directly from the browser version for quick testing.