客户用官方mindspore的yolov5代码仓源码训练自己的业务数据,训练好的ckpt导出onnx,在atlas200 dk a2上用atc转成om,然后用om替换atlas200 dk里的yolo示例里的om,推理报错,报错信息如下,请求帮助
/usr/local/miniconda3/lib/python3.9/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")
[INFO] acl init success
[INFO] open device 0 success
[INFO] load model model.om success
[INFO] create model description success
[ERROR] Check i:0 name:x in size:4915200 needsize:1228800 not match
[ERROR] Check InVector failed ret:-1
Traceback (most recent call last):
File "/home/HwHiAiUser/samples/notebooks/01-yolov5/infer.py", line 157, in <module>
infer_image(img_path, model, labels_dict, cfg)
File "/home/HwHiAiUser/samples/notebooks/01-yolov5/infer.py", line 64, in infer_image
output = model.infer([img])[0]
File "/usr/local/miniconda3/lib/python3.9/site-packages/ais_bench/infer/interface.py", line 162, in infer
return self.run(inputs, out_array=True)
File "/usr/local/miniconda3/lib/python3.9/site-packages/ais_bench/infer/interface.py", line 94, in run
outputs = self.session.run(self.outputs_names, inputs)
RuntimeError: [-1][ACL: general failure]
[INFO] unload model success, model Id is 1
[INFO] end to destroy context
[INFO] end to reset device is 0
[INFO] end to finalize acl
客户用官方mindspore的yolov5代码仓源码训练自己的业务数据,训练好的ckpt导出onnx,在atlas200 dk a2上用atc转成om,然后用om替换atlas200 dk里的yolo示例里的om,推理报错,报错信息如下,请求帮助
/usr/local/miniconda3/lib/python3.9/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")
[INFO] acl init success
[INFO] open device 0 success
[INFO] load model model.om success
[INFO] create model description success
[ERROR] Check i:0 name:x in size:4915200 needsize:1228800 not match
[ERROR] Check InVector failed ret:-1
Traceback (most recent call last):
File "/home/HwHiAiUser/samples/notebooks/01-yolov5/infer.py", line 157, in <module>
infer_image(img_path, model, labels_dict, cfg)
File "/home/HwHiAiUser/samples/notebooks/01-yolov5/infer.py", line 64, in infer_image
output = model.infer([img])[0]
File "/usr/local/miniconda3/lib/python3.9/site-packages/ais_bench/infer/interface.py", line 162, in infer
return self.run(inputs, out_array=True)
File "/usr/local/miniconda3/lib/python3.9/site-packages/ais_bench/infer/interface.py", line 94, in run
outputs = self.session.run(self.outputs_names, inputs)
RuntimeError: [-1][ACL: general failure]
[INFO] unload model success, model Id is 1
[INFO] end to destroy context
[INFO] end to reset device is 0
[INFO] end to finalize acl