文档
注册

模型可视化

对于原始模型文件(.pb、.onnx、.prototxt)或转换成功的.om模型文件,可以在MindStudio界面呈现其网络拓扑结构,并可以查看模型所使用的算子。下文中以.om模型文件为例展示模型可视化功能。

前提条件

如果Linux环境为aarch64架构,且系统中的glibc版本小于2.29,则无法使用MindStudio的模型可视化功能,处理方法请参考glibc版本过低导致无法使用模型可视化功能

操作步骤

  1. 依次单击菜单栏Ascend > Model Visualizer,或在工具栏选择,在弹出窗口中选择要可视化的模型,例如此处的resnet50模型文件。
    • 在Linux环境上使用模型可视化功能时,如图1所示。
      图1 选择要打开的模型(Linux)
    • 在Windows环境上使用模型可视化功能时,如图2所示。
      图2 选择要打开的模型(Windows)

      查看.prototxt模型文件时,需要保证.prototxt模型文件与其相关的.caffemodel权重文件在同一目录下。

  2. 鼠标左键双击resnet50.om模型文件或者单击“OK”,打开模型可视化界面如图3 模型可视化界面所示。
    图3 模型可视化界面

可视化界面说明

  • 查看算子信息

    图3区域一中展示了模型文件中所有的算子,单击某层算子,该层算子会出现绿色选中框,区域三会展示该算子的详细信息,包括算子名称、算子输入、输出等信息。

    区域2会展示该网络模型的整体结构,包括区域三中展示的算子在整体网络结构中的位置,即图中的蓝色框选择位置。

    • 上/下滑动鼠标滚轮,可以控制模型可视化区域的上/下移动。
    • 按住Ctrl键+上下滑动鼠标滚轮,可以实现模型可视化区域的放大/缩小。
  • 查看算子输出维度和shape信息

    图3区域一中还展示了每一层算子输出的shape信息,如每一层算子连接线中间的1,224,224,4等shape信息。

  • 搜索算子

    图3区域四中的“Find”输入框中输入算子名称,下方搜索区域会列出相关的算子。选择其中一个算子,区域一中的网络拓扑结构中相应算子会显示绿色选中框,区域三会展示该算子的详细信息。如图4所示。

    图4 模型可视化界面的搜索功能
  • 搜索算子内部信息

    区域中某层算子,在区域三图标后的输入框中输入想要查询的信息,比如算子的输入输出、属性等信息,然后单击图标,如果匹配到相关信息,则区域三中相关信息会高亮显示,否则会在“Output”窗口提示“Value not found”

  • 查看模型输出节点信息

    当模型转换时通过Select节点设置了输出算子,则在.om模型文件可以看到Select算子的输出直接作为模型的输出,可以单击相应的算子,查看其输入信息。

  • 查看模型输出节点的数据类型

    如果模型转换时设置了Select节点的输出类型,对应执行转换章节“Model Information”页签中Output Nodes区域下方节点的Data Type类型,模型转换完毕,用户可以单击对应的节点,查看不同输入节点的数据类型。

    如果模型转换时只设置了“Model Information”页签中的“Output Type”参数取值,没有Select节点,则模型转换完毕,单击对应的节点,查看输入数据类型即可。

    dtype为DT_FLOAT,表示数据类型为FP32,dtype为DT_FLOAT16表示数据类型为FP16,dtype为DT_UINT8表示数据类型为UINT8。

其他操作

Linux环境aarch64架构

UbuntuOS:https://mindstudio-sample.obs.cn-north-4.myhuaweicloud.com/glibc/euler-os/glibc-2.29.tar.gz

EulerOS:https://mindstudio-sample.obs.cn-north-4.myhuaweicloud.com/glibc/euler-os/glibc-2.29.tar.gz

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词