ONNX模型在线查看工具

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
  • 输出集:包含两个张量 scoresboxes ,数据类型均为 float32  。从名称可以推断分别代表检测到的区域的得分( 1 x 4420 x 2)和包围盒( 1 x 4420 x4)。

3、查看节点属性

也可以选中计算图种的任意节点查看其节点属性:

4、使用菜单

点击打开左上角的汉堡菜单,其中包括一些辅助的工具,例如导出图像、缩放等:

其中一个比较好用的工具是切换计算图的水平/垂直显示。初始打开模型时默认是垂直显示,我们点击一下切换到水平显示:


链接:Netron 中文版 - ONNX模型在线查看