Netron 是一款跨平台机器学习模型可视化工具,只需上传 ONNX 文件或 Prototxt 即可可视化模型的结构。
有两种方法可以使用:
1)本地安装,见Github 上README;
2)使用浏览器打开Netron:https://netron.app/ 。
打开模型即可。然后就可以看见模型结构。
单击每一层会看见如下图,帮助更好理解模型。
Netron 支持 ONNX、TensorFlow Lite、Caffe、Keras、Darknet、PaddlePaddle、ncnn、MNN、Core ML、RKNN、MXNet、MindSpore Lite、TNN、Barracuda、Tengine、CNTK、TensorFlow.js、Caffe2 和 UFF。