飞腾D2000 + Ascend 310 AI处理器推理RTDETR时Execute model inference failed
收藏回复举报
飞腾D2000 + Ascend 310 AI处理器推理RTDETR时Execute model inference failed
t('forum.solved') 已解决
新人帖
发表于2025-02-19 15:42:45
0 查看

设备:飞腾D2000 + Ascend 310 AI处理器

版本信息

操作系统:Ubuntu 18.04.6 LTS

cann版本: package_name=Ascend-cann-toolkit     version=6.3.RC2 

代码:RTDETR代码

开发内容:onnx转om模型后c++推理

问题描述:

python tools/export_onnx.py --config configs/rtdetr/rtdetr_r18vd_6x_coco.yml -r ../models/pytorch/rtdetr_r18vd_dec3_6x_coco_from_paddle.pth --check

将rtdetr_r18vd_dec3_6x_coco_from_paddle.pth官方模型转onnx模型,onnx模型能正常推理,om模型转换也没报错。网络是两个输入三个输出。模型可以正常加载,但推理报错。

-- Create instance of Resource RTDETRAscendDetector 

[INFO]  Acl init success 

[INFO]  Open device 0 success 

[INFO]  Use default context currently 

[INFO]  load model ../model/RTDETR-COCO//model.om success 

[INFO]  create model description success 

[INFO]  model input width 640, input height 640 

aclmdlGetOutputDims: 1 300 640 3 

[INFO]  create model output success 

--Loaded Net-Config with ../model/RTDETR-COCO/ 

507011 

[ERROR]   Execute model inference failed, modelId is 1 

附上ascend文件夹中的.log和使用的onnx文件:

链接: https://pan.baidu.com/s/1PXgJ8glW3KMRclAVo8sO9g 提取码: pi5k 复制这段内容后打开百度网盘手机App,操作更方便哦

本帖最后由 匿名用户2025/02/19 17:12:22 编辑

我要发帖子