yolov8mpt模型转为om后进行推理,报错显示输入的尺寸不对
收藏回复举报
yolov8mpt模型转为om后进行推理,报错显示输入的尺寸不对
发表于2023-09-25 16:17:03
0 查看

根据https://gitee.com/ascend/modelzoo-GPL/tree/master/built-in/ACL_Pytorch/Yolov8_for_PyTorch 使用om_info.py文件里面对om模型推理的部分,推理报错

[ERROR] Check i:0 name:images in size:3240 needsize:4915200 not match

[ERROR]check Invector failed ret:-1

Traceback (most recent call last):

File“/home/HwHiAiUser/ads_detect/test.py”, line 51,in <module>

main().

File“/home/HwHiAiUser/ads_detect/test.py”,line 47,

File“/home/HwHiAiUser/ads_detect/test.py”,line 36,in val

preds = om model.infer(image)

File"/usr/local/python3.9.2/lib/python3.9/site-packages/ais bench/infer/interface.py”, line 162,in infer

return self.rn(inputs, out arraw-True]

File "/usr/local/python3.9.2/lib/python3.9/site-packages/ais bench/infer/interface.py", line 94,in run

outouts =self.session.run(self.outputs_names, inputs)

RuntimeError:[-1][ACL: general failure]

本帖最后由 匿名用户2023/09/25 18:55:52 编辑

我要发帖子