原始模型文件或离线模型转成json文件

场景介绍

如果用户不方便查看原始模型或离线模型的参数信息时，可以将原始模型或离线模型转成json文件进行查看。

转换方法

本章节以TensorFlow框架ResNet-50网络模型为例进行演示，单击Link，根据页面提示获取ResNet50网络模型文件（*.pb）。

• 原始模型文件转json文件

  命令示例如下：

  atc --mode=1 --om=$HOME/module/resnet50_tensorflow*.pb  --json=$HOME/module/out/tf_resnet50.json  --framework=3

  • --mode：运行模式，1表示原始模型文件或离线模型转json，此处特指原始模型文件转json。
  • --om：指定ResNet-50网络模型文件所在路径。
  • --json：转换为json格式的文件路径和文件名。
  • --framework：原始框架类型，3表示TensorFlow。
• 离线模型转json文件
  该场景的前提是用户根据开源框架的TensorFlow网络模型转换成离线模型已经得到了.om离线模型文件，命令示例如下：

  atc --mode=1 --om=$HOME/module/out/tf_resnet50.om  --json=$HOME/module/out/tf_resnet50.json

  • --mode：运行模式，1表示原始模型文件或离线模型转json，此处特指离线模型文件转json。
  • --om：指定离线模型文件所在路径。

1

ATC run success, welcome to the next use.

成功执行命令后，在--json参数指定的路径下，可查看转换后的json文件信息，如下为部分json片段：

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

{
  "node": [
    {
      "attr": [
        {
          "key": "shape",
          "value": {
            "shape": {
              "dim": [
                {
                  "size": 1
                },
                {
                  "size": 224
                },
                {
                  "size": 224
                },
                {
                  "size": 3
                }
              ]
            }
          }
        },
        {
          "key": "dtype",
          "value": {
            "type": "DT_FLOAT"
          }
        }
      ],
      "name": "Placeholder",
      "op": "Placeholder"
    },
...