原始模型文件或离线模型转成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的操作。
- 原始模型转成离线模型,命令示例如下:
atc --model=$HOME/module/resnet50_tensorflow*.pb --framework=3 --output=$HOME/module/out/tf_resnet50 --soc_version=<soc_version>
- 离线模型转成json文件,命令示例如下:
atc --mode=1 --om=$HOME/module/out/tf_resnet50.om --json=$HOME/module/out/tf_resnet50.json
成功执行命令后,在--json参数指定的路径下,可查看转换后的json文件信息。
- 原始模型转成离线模型,命令示例如下:
父主题: 初级功能