使用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/目录下。
使用示例
- 以DDK安装用户登录DDK所在服务器。
- 设置环境变量:
export LD_LIBRARY_PATH=DDK安装目录/ddk/uihost/lib
- 进入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解决。
- (可选)如果模型转换过程中指定了输出节点(即使用了--out_nodes参数),转换成.om模型后无法查看最后一层算子的输出信息,则可以将.om模型转换成json格式后查看,使用命令为:
omg --mode=1 --om=/home/username/test/caffe_resnet18.om --json=/home/username/test/out/resnet.json