# onnx2om

**Repository Path**: sunqiangAAA/onnx2om

## Basic Information

- **Project Name**: onnx2om
- **Description**: ONNX模型转om
- **Primary Language**: Unknown
- **License**: MulanPSL-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2024-09-05
- **Last Updated**: 2024-09-27

## Categories & Tags

**Categories**: Uncategorized

**Tags**: 华为昇腾, 香橙派, om, onnx, CANN

## README

# ONNX转om脚本

## 说明：

香橙派AiPro华为昇腾NPU架构om转化（CANN7.0.0）

ATC常用参数（详细见[参考1](https://www.hiascend.com/document/detail/zh/Atlas200IDKA2DeveloperKit/23.0.RC1/Appendices/tmuacop/atctool_000041.html)）：

- --model: 原始模型文件路径与文件名。（必须）
- --framework: 原始网络模型框架类型, ONNX为5。（必须）
  - 0：Caffe
  - 1：MindSpore
  - 3：TensorFlow
  - 5：ONNX
- --output: 存放转换后的离线模型的路径以及文件名。（必须）
- --input_format: 输入数据格式，默认NCHW。（必须）
- --input_shape: 模型输入数据的shape， 例如"images:1,3,640,640"（其中输入名images需要再netron中查看）。（必须）
- --soc_version: 模型转换时指定芯片版本（npu-smi info 查看对应版本）。（必须）
- --input_fp16_nodes: 指定输入数据类型为FP16的输入节点名称。
- --output_type: 指定网络输出数据类型或指定某个输出节点的输出类型。

## 环境：

1、环境拉取

```shell
sudo docker pull sunqiang188/onnx2om:latest
```

2、环境运行

```shell
sudo docker run -it --name onnx2om -v <本机onnx模型所在路径>:/home sunqiang188/onnx2om:latest /bin/bash
```

## 运行:

如果已经退出docker环境，需要进入（Docker外部）：

```shell
sudo docker exec -it onnx2om /bin/bash
```

运行ATC（Docker内部）：

```shell
cd /home
atc --model=smoke.onnx --framework=5 --output=smoke_bs1 --input_format=NCHW --input_shape="images:1,3,640,640" --log=error --soc_version=Ascend310B4 --input_fp16_nodes "images" --output_type=FP16
```

## 参考：

1、ATC参数说明：https://www.hiascend.com/document/detail/zh/Atlas200IDKA2DeveloperKit/23.0.RC1/Appendices/tmuacop/atctool_000041.html

2、环境搭建：https://www.hiascend.com/document/detail/zh/CANNCommunityEdition/700alpha002/softwareinstall/instg/instg_0028.html