下载
EN
注册

将ONNX模型转换为OM模型

导出ONNX模型导出的“resnet50.onnx”模型,用CANN提供的ATC工具将其转换为昇腾AI处理器能识别的OM模型。

1
atc --model=resnet50.onnx --framework=5 --output=resnet50 --input_shape="actual_input_1:1,3,224,224"  --soc_version=<soc_version>

atc命令中各参数的含义如下:

  • --model:ResNet50的ONNX模型文件存储路径。
  • --framework:原始框架类型,5表示ONNX。
  • --output:离线OM模型的路径以及文件名,此处文件名配置为resnet50,直接存储在当前目录下。

    请注意,请记录保存该OM模型文件的路径,后续推理应用程序需要使用。

  • --input_shape:执行推理时模型输入数据的shape。
  • --soc_version:昇腾AI处理器的型号。

    请在服务器种执行npu-smi info命令进行查询,在查询到的“Name”前增加Ascend信息,例如“Name”对应取值为xxxyy,实际配置的<soc_version>值为Ascendxxxyy