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.