以下介绍通过ATC模型转换工具获取离线模型的操作步骤,更多操作请参见《ATC离线模型编译工具用户指南》。
例如:$HOME/module/resnet50_tensorflow*.pb
atc --model=$HOME/module/resnet50_tensorflow*.pb --framework=3 --output=$HOME/module/out/tf_resnet50 --soc_version=<soc_version>
1 | ATC run success
|
成功执行命令后,在--output参数指定的路径下可查看离线模型(如:tf_resnet50.om)。
atc --mode=1 --om=$HOME/module/out/tf_resnet50.om --json=$HOME/module/out/tf_resnet50.json
若提示如下信息,则说明转换json文件成功。
1 | ATC run success
|
成功执行命令后,在--json参数指定的路径下可查看转换后的json文件。