DEIMv2 模型在昇腾上的Ascend适配
收藏回复举报
DEIMv2 模型在昇腾上的Ascend适配
t('forum.solved') 已解决
新人帖
发表于2025-10-09 17:32:24
0 查看

求助:

我已经在onnx和C++的上面适配了最新的DEIMv2模型(https://github.com/Choise-ieee/DEIMv2_CPP)。

image.png

我想在昇腾平台上实现迁移,发现并不能成功,

我的在C++和onnx上跑通的onnx模型地址可下载地址:https://drive.google.com/file/d/1nPKDHrotusQ748O1cQXJfi5wdShq6bKp/view?usp=drive_link

然后我利用ATC工具,将其转为昇腾的om模型,命令为atc --model=deimv2_dinov3_m_coco.onnx --framework=5 --output=ascend_model --soc_version=Ascend310B4 --input_shape="images:1,3,640,640;orig_target_sizes:1,2" --input_format=NCHW --insert_op_conf=aipp.cfg --optypelist_for_implmode="Abs" --op_select_implmode=high_precision --precision_mode=force_fp16 --disable_reuse_memory=1 --output_type=FP16

设备是310B4,转出的om文件,下载地址为:https://drive.google.com/file/d/1zu36UoUq4yFc9cpyc7VjS7pCUTYwMBZ7/view?usp=drive_link

其中配套的aipp.cfg,和最终的C++程序,下载地址是:https://drive.google.com/file/d/1Bx_0z3O-fv-17p_nXNoXAJF9FF5JbkkK/view?usp=drive_link

最终运行失败的结果为

cke_19180.jpeg

不知道是何原因,个人想实现DEIMv2 模型在昇腾上的Ascend适配,并最后开源,所以求赐教,谢谢

本帖最后由 匿名用户2025/10/10 09:42:08 编辑

我要发帖子