模型可视化

对于原始模型文件(.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 模型可视化界面

可视化界面说明

其他操作

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