目录

可视化工具

概述

Netron是一个神经网络模型可视化工具，支持许多主流AI框架模型的可视化。Netron 5.1.6版本开始支持.om模型可视化。如下图所示，使用Netron工具加载.om模型后，可以展示模型的拓扑结构、图、节点的信息等。

功能描述

支持加载.om模型。
支持展示拓扑结构和数据流shape。
支持查看模型的format、input和output等参数。
支持查看编译后模型的子图和算子设备信息。
支持查看节点的NODE PROPERTIES、ATTRIBUTES、INPUTS和OUTPUTS等信息。
支持保存可视化结果导出为图片。

使用可视化工具

[h2]安装工具

下载最新的Netron。
安装Netron。
- macOS: 下载.dmg文件或者执行brew cask install netron。
- Linux: 下载.AppImage文件或者执行snap install netron。
- Windows: 下载.exe文件或者执行winget install netron。
- Python服务器：执行pip install netron安装Netron，然后通过netron [FILE]或netron.start('[FILE]')加载模型。
- 浏览器：无需安装，直接打开网页端Netron可使用。
安装完成后，将模型拖入窗口即可打开。

[h2]查看子图

对于编译后有子图的模型，可按照如下操作查看。

将编译后的模型拖入Netron工具，即可打开。
点击子图节点，在右侧查找"ATTRIBUTES->subgraph"，点击"subgraph"的属性值。
查看子图节点的NODE PROPERTIES、ATTRIBUTES、INPUTS和OUTPUTS等信息。
点击左上角箭头，返回主图。