# acl_om_yolov5

**Repository Path**: don-quixote-der/acl_om_yolov5

## Basic Information

- **Project Name**: acl_om_yolov5
- **Description**: 在昇腾平台上使用PyACL调用 yolov5进行推理代码
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 6
- **Forks**: 3
- **Created**: 2023-10-17
- **Last Updated**: 2025-07-16

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

## 昇腾Ascend 离线推理 Yolov5 示例脚本



同时提供onnx推理代码与om推理代码,采用技术栈为OM离线模型与PyACL

### 环境安装

onnx推理需要安装依赖,由于opencv原因，需要重新卸载opencv-python opencv-python-headless ，再安装opencv-python-headless

```shell
pip install -r  requirment.txt

pip uninstall -y opencv-python

pip uninstall -y opencv-python-headless

pip install opencv-python-headless
```

下载yolov5s官方onnx 链接：https://pan.baidu.com/s/19m7pCDK_O3qPPp7d8AIeuw?pwd=fy1z  提取码：fy1z 

### 使用onnx直接在cpu上直接推理

![image-20231018152150652](media/image-20231018152150652.png)

### 使用atc进行转换

```shell
atc --framework=5 \
    --model=yolov5s.onnx \
    --input_shape="images:1,3,640,640"  \
    --output="yolov5s" \
    --enable_small_channel=1 \
    --log=error \
    --soc_version=Ascend910B \
    --op_select_implmode=high_precision \
    --output_type=FP32

```

![image-20231017144132940](media/image-20231017144132940.png)

### 运行推理

python predict_om.py

![image-20231017150944659](media/image-20231017150944659.png)

### 推理结果

![image-20231017151023773](media/image-20231017151023773.png)