Yolov5s自己训练的模型用atc命令转om格式后在Atlas 200I DK A2板端推理运行不了
收藏回复举报
Yolov5s自己训练的模型用atc命令转om格式后在Atlas 200I DK A2板端推理运行不了
t('forum.solved') 已解决
发表于2024-08-21 10:23:22
0 查看

【用户类型】:企业

【昇腾产品型号】:Atlas 200I DK A2

【版本信息】: 

--操作系统版本:Ubuntu 22.04 LTS aarch64 

--固件驱动版本:npu-smi 23.0.rc3     Version: 23.0.rc3

--CANN版本:CANN 7.0.rc1 

--Pytorch版本:1.13.0 

--Python版本:Python 3.9.2

【模型权重推理脚本】: 使用yolov5的官方权重yolv5s.pt,基于自己的数据和类别训练的模型

模型和推理demo链接:通过百度网盘分享的文件:Atlas开发板模型转换及推理 

链接:https://pan.baidu.com/s/1o8Jop4_UdBhO7WBWUHAS_Q  

提取码:bovr

【开发内容】:使用yolov5做一个电焊作业监管设备,部署在Atlas 200I DK A2 开发版推理

【问题描述】:使用yolov5的export.py脚本将其转为onnx格式,然后上传到开发板使用atc命令转换om格式后,使用开发板预置于文件夹/root/yolo_sdk_python_sample里面的yolov5 demo测试,推理报错,模型和代码分享在链接了。

报错如下:

cke_27513.png

我要发帖子