Netron中文版是一个功能强大的在线工具,专门用于ONNX、TF Lite等多种格式的神经网络模型的可视化查看。
Netron中文版的主要特性如下:
- 支持多种模型格式,包括 ONNX、TensorFlow、Keras、Caffe 和 PyTorch等
- 提供直观的图形界面,帮助用户浏览模型结构、查看各层信息、检查参数,并理解数据在网络中的流动。
- 可作为网页应用、桌面应用使用,也可以嵌入到其他工具中。
1、打开本地ONNX模型
点击这里打开Netron中文版首页:
点击【打开模型文件】按钮,即可打开本地的AI/ML模型模型。如果你手头没有,可以下载这个常用的人脸检测模型 version-RFB-320.onnx
进行练习(下载链接:百度网盘 或 Github )。打开后如下图所示:
2、查看模型属性
点击左下角的第一个图标,即可查看模型的主要属性:
对于需要调用模型的开发者而言,最重要的是输入/输出部分的信息。例如对于示例种的version-RFB-320.onnx
模型,可以看到:
- 输入集:仅包含一个名为
input
的张量,形状为1 x 3 x 240 x 321
,数据类型为float32
。 - 输出集:包含两个张量
scores
和boxes
,数据类型均为float32
。从名称可以推断分别代表检测到的区域的得分(1 x 4420 x 2
)和包围盒(1 x 4420 x4
)。
3、查看节点属性
也可以选中计算图种的任意节点查看其节点属性:
4、使用菜单
点击打开左上角的汉堡菜单,其中包括一些辅助的工具,例如导出图像、缩放等:
其中一个比较好用的工具是切换计算图的水平/垂直显示。初始打开模型时默认是垂直显示,我们点击一下切换到水平显示:
原文链接:ONNX模型在线查看工具 - BimAnt