昇腾社区首页
中文
注册

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

场景介绍

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

转换方法

本章节以Caffe框架ResNet-50网络模型为例,进行演示,参见1获取原始模型文件。

  • 原始模型文件转json文件

    该场景下--om参数需要指定为原始模型文件,命令示例如下:

    atc --mode=1 --om=$HOME/module/resnet50.prototxt  --json=$HOME/module/out/caffe_resnet50.json  --framework=0
  • 离线模型转json文件

    该场景下需要先将原始模型转成离线模型,然后再执行离线模型转成json的操作。

    1. 原始模型转成离线模型,命令示例如下:
      atc --model=$HOME/module/resnet50.prototxt --weight=$HOME/module/resnet50.caffemodel --framework=0 --output=$HOME/module/out/caffe_resnet50 --soc_version=<soc_version>  
    2. 离线模型转成json文件,命令示例如下:
      atc --mode=1 --om=$HOME/module/out/caffe_resnet50.om  --json=$HOME/module/out/caffe_resnet50.json
    关于参数的详细解释以及使用方法请参见参考。若提示如下信息,则说明转换成功,若模型转换失败,则请参见错误码参考进行定位。
    ATC run success

    成功执行命令后,在--json参数指定的路径下,可查看转换后的json文件信息。

    关于参数的详细解释以及使用方法请参见参考