【成都AICC】推理CDUNet模型,报错:CheckUserAndModelSize: ErrorNo: 145000(Parameter invalid.)
收藏回复举报
【成都AICC】推理CDUNet模型,报错:CheckUserAndModelSize: ErrorNo: 145000(Parameter invalid.)
t('forum.solved') 已解决
发表于2024-09-12 17:43:24
0 查看

使用CDUNet模型,onnx转om后,自行编写推理代码,推理报错:CheckUserAndModelSize: ErrorNo: 145000(Parameter invalid.)

看报错信息是输入不匹配,该模型有两个输入,1,3,544,544,检查了模型的输入和图片尺寸都是一致的。

用的6.0.1的cann,不清楚和cann版本有没有关系。

image.png

ATC转换命令:atc --model=best_0909.onnx --framework=5 --output=pb_310 --input_shape="input1:1,3,544,544;input2:1,3,544,544" --soc_version=Ascend310

模型结构:

image.png

推理代码:

image.png

image.png

我要发帖子