文档
注册
评分
提单
论坛
小AI

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

场景介绍

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

转换方法

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

  • 原始模型文件转json文件

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

    atc --mode=1 --om=$HOME/module/resnet50_tensorflow*.pb  --json=$HOME/module/out/tf_resnet50.json  --framework=3
  • 离线模型转json文件

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

    1. 原始模型转成离线模型,命令示例如下:
      atc --model=$HOME/module/resnet50_tensorflow*.pb --framework=3 --output=$HOME/module/out/tf_resnet50 --soc_version=<soc_version>   
    2. 离线模型转成json文件,命令示例如下:
      atc --mode=1 --om=$HOME/module/out/tf_resnet50.om  --json=$HOME/module/out/tf_resnet50.json
    关于参数的详细解释以及使用方法请参见参数说明。若提示如下信息,则说明转换成功,若模型转换失败,则请参见故障处理进行定位。
    ATC run success

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

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

当前产品无相关内容

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