Netron是一款专为机器学习模型设计的可视化工具,版本5.8.2提供直观界面解析各类框架(如TensorFlow、PyTorch)的模型结构。
Netron 是一个用于查看神经网络、深度学习和机器学习模型的工具。它支持多种格式,包括 ONNX、Keras、TensorFlow Lite、Caffe、Darknet、Core ML、MNN、MXNet、ncnn、PaddlePaddle、Caffe2、Tengine、TNN 和 RKNN。此外,还为 TensorFlow、PyTorch 以及 scikit-learn 提供实验性支持。
该工具可在 macOS、Windows、Linux 平台上使用,并且有 Python Server 和浏览器版本可供选择。它提供大量样本模型文件以供下载或直接在浏览器中打开。
以下是 Netron 支持的一些具体示例:
- ONNX:squeezenet 模型
- CoreML:exermote 模型
- Darknet:yolo 模型
- Keras:移动网络模型
- MXNet:inception_v3 模型
- TensorFlow:国际象棋机器人模型