【AI】Netron

长期更新，建议收藏关注！

友情链接
onnx

Netron 是一个神经网络、深度学习和机器学习模型的浏览器。它可以为模型的架构生成具有描述性的可视化。

支持包括：
支持ONNX, TensorFlow Lite, Caffe, Keras, Darknet, PaddlePaddle, ncnn, MNN, Core ML, RKNN, MXNet, MindSpore Lite, TNN, Barracuda, Tengine, CNTK, TensorFlow.js, Caffe2, UFF.
以及：
实验性支持PyTorch, TensorFlow, TorchScript, OpenVINO, Torch, Vitis AI, kmodel, Arm NN, BigDL, Chainer, Deeplearning4j, MediaPipe, MegEngine, ML.NET and scikit-learn

整体使用特征图的方式进行对模型的描述，并在旁边写出对应的参数。如图所示

安装

网页使用：https://netron.app
Github下载：https://github.com/lutzroeder/netron

macOS：下载 .dmg 文件或运行 brew install --cask netron
Linux：下载 .AppImage 文件或运行 snap install netron
Windows：下载 .exe 安装程序或运行 winget install -s winget netron
Python 服务器：运行 pip install netron 和 netron [FILE] 或 netron.start(‘[FILE]’) 。

使用的话直接把模型喂给它，直接拖入模型即可，并可以方便保存为图片（PNG或SVG格式），比如：
YOLO v3 的.cfg格式文件，Resnet 34 .pth格式预训练模型（PyTorch），PNASNet-5 .ckpt格式预训练模型 （TensorFlow），VGG 16 train_val .prototxt格式文件（Caffe）

参考资料

本文仅供个人学习使用，免费分享。

1. https://mp.weixin.qq.com/s/fJlcixijWr3XXwL8Nj398A