模型可视化

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

操作步骤

  1. 依次单击菜单栏Ascend > Model Visualizer,或在工具栏选择,在弹出窗口中选择要可视化的模型,例如此处的resnet50模型文件。

    • 在Linux环境上使用模型可视化功能时,如图1所示。
      图1 选择要打开的模型(Linux)
    • 在Windows环境上使用模型可视化功能时,如图2所示。
      图2 选择要打开的模型(Windows)

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

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

    • 图3为模型可视化优化前的排版和结构。
      图3 模型可视化界面(优化前)
    • 图4为模型可视化优化后的排版和结构。
      图4 模型可视化界面(优化后)
      以root用户安装node.js工具和dagre-d3插件后,可以优化模型可视化的排版布局效果,使整个网络拓扑结构更加清晰有条理,安装方式如下所示:
      • 安装node.js工具
        • ubuntu操作系统:apt install nodejs
        • CentOS、EulerOS和麒麟OS:
          1. 如在官网下载软件包。

            例如,下载node-v8.10.0-linux-arm64.tar.gz

          2. 解压软件包。
          3. 配置环境变量:export NODE_HOME=<软件包解压目录>export PATH=<软件包解压目录>/bin:$PATH
      • 安装dagre-d3插件
        • npm install dagre-d3 -g
        • 配置环境变量:export NODE_PATH=<软件包安装目录>

可视化界面说明