在华为云官网下载Yolov5.onnx模型进行om模型转换后进行推理结果与烧写镜像自带的im模型推理结果差别很大
收藏回复举报
在华为云官网下载Yolov5.onnx模型进行om模型转换后进行推理结果与烧写镜像自带的im模型推理结果差别很大
t('forum.solved') 已解决
发表于2023-09-14 16:05:25
0 查看

1. 在华为云官网下载ONNX模型——

https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Atlas%20200I%20DK%20A2/DevKit/samples/23.0.RC1/base-samples/notebook-demo-datasets/01-yolov5/yolov5s.onnx

2. 传输到开发板上使用atc工具进行om转换,转换命令为

atc --model=yolov5s_1.onnx --framework=5 --output=yolo1 --input_format=NCHW --input_shape="images:1,3,640,640" --log=error --soc_version=Ascend310B1 

3. 使用烧写镜像后自带的jupyter文件进行推理,将其中yolo.om替换为转换后的yolo1.om文件,推理结果如下

Snipaste_2023-09-14_16-02-11.png

而板子自带的yolo.om文件推理结果为:

2.png

请问是atc命令出了问题吗?转换时有什么额外的参数没有指定吗?

我要发帖子