NSDT工具推荐: Three.js AI纹理开发包 - YOLO合成数据生成器 - GLTF/GLB在线编辑 - 3D模型格式在线转换 - 可编程3D场景编辑器 - REVIT导出3D模型插件 - 3D模型语义搜索引擎 - AI模型在线查看 - Three.js虚拟轴心开发包 - 3D模型在线减面 - STL模型在线切割 - 3D道路快速建模
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、使用菜单
点击打开左上角的汉堡菜单,其中包括一些辅助的工具,例如导出图像、缩放等:
其中一个比较好用的工具是切换计算图的水平/垂直显示。初始打开模型时默认是垂直显示,我们点击一下切换到水平显示: