更新时间:2021-03-18 GMT+08:00
分享

使用OMG工具转换模型

准备动作

  • 环境准备

    本手册介绍命令行方式的模型转换,如果用户想通过Mind Studio界面进行模型转换,则请参见《Ascend 310 Mind Studio基本操作》

    本手册以DDK独立安装为例进行说明。

    如果DDK安装用户为HwHiAiUser,日志未输出到屏幕或者DDK所在服务器操作系统以及架构为Arm(aarch64),模型转换耗时较长,则可以分别参见DDK安装用户为HwHiAiUser时,日志未输出到屏幕DDK所在服务器操作系统以及架构为Arm(aarch64),模型转换耗时较长解决。

  • 以DDK安装用户将模型转换过程中使用到的模型文件(*.prototxt)、权重文件(*.caffemodel)等上传到DDK所在服务器/home/username/test/目录下。

使用示例

  1. 以DDK安装用户登录DDK所在服务器。
  2. 设置环境变量:

    export  LD_LIBRARY_PATH=DDK安装目录/ddk/uihost/lib

  3. 进入DDK安装目录“ddk/uihost/bin”下获取omg工具。执行以下命令生成模型文件。(如下命令中使用的目录以及文件均为样例,请以实际为准)

    ./omg --model=/home/username/test/resnet18.prototxt --weight=/home/username/test/resnet18.caffemodel --framework=0 --output=/home/username/test/out/caffe_resnet18 

    成功执行命令后,在output参数指定的路径下,可查看模型文件(如:caffe_resnet18.om)。

    如果您直接复制示例中的命令,由于PDF文档格式的限制,超过单行的命令,会自动换行,因此您需要手动将多行命令合并成一行,参数之间以空格分割。

    模型转换过程中如果有如下错误提示信息“It is recommended to convert layers-structure to layer-structure by caffe tool”“Type XXX unsupported”,则请参见FAQ解决。

  4. (可选)如果模型转换过程中指定了输出节点(即使用了--out_nodes参数),转换成.om模型后无法查看最后一层算子的输出信息,则可以将.om模型转换成json格式后查看,使用命令为:

    omg --mode=1 --om=/home/username/test/caffe_resnet18.om  --json=/home/username/test/out/resnet.json

分享:

    相关文档

    相关产品