您好,我在参考https://gitee.com/ascend/modelzoo-GPL/tree/master/built-in/ACL_Pytorch/Yolov8_for_PyTorch#/ascend/modelzoo-GPL/blob/master/built-in/ACL_Pytorch/Yolov8_for_PyTorch/文档进行Yolov8板端部署,目前转模型已经成功,但在执行
这条语句的时候报错:[INFO] acl init success
[INFO] open device 0 success
[INFO] load model ../yolov8n_bs1.om success
[INFO] create model description success
Class Images Instances Box(P R mAP50 mAP50-95): 0%| | 0/5000 [00:00<?, ?it/s][ERROR] Check i:0 name:images in size:2457600 needsize:4915200 not match
[ERROR] Check InVector failed ret:-1
Class Images Instances Box(P R mAP50 mAP50-95): 0%| | 0/5000 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/root/hanyi/Yolov8_for_PyTorch/ultralytics/../om_infer.py", line 467, in <module>
main()
File "/root/hanyi/Yolov8_for_PyTorch/ultralytics/../om_infer.py", line 463, in main
val(input_args)
File "/root/hanyi/Yolov8_for_PyTorch/ultralytics/../om_infer.py", line 433, in val
preds = om_model.infer([batch['img']])
File "/root/hanyi/Yolov8_for_PyTorch/venv/lib/python3.9/site-packages/ais_bench/infer/interface.py", line 162, in infer
return self.run(inputs, out_array=True)
File "/root/hanyi/Yolov8_for_PyTorch/venv/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
请问该如何解决?
这条语句的时候报错:[INFO] acl init success
[INFO] open device 0 success
[INFO] load model ../yolov8n_bs1.om success
[INFO] create model description success
Class Images Instances Box(P R mAP50 mAP50-95): 0%| | 0/5000 [00:00<?, ?it/s][ERROR] Check i:0 name:images in size:2457600 needsize:4915200 not match
[ERROR] Check InVector failed ret:-1
Class Images Instances Box(P R mAP50 mAP50-95): 0%| | 0/5000 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/root/hanyi/Yolov8_for_PyTorch/ultralytics/../om_infer.py", line 467, in <module>
main()
File "/root/hanyi/Yolov8_for_PyTorch/ultralytics/../om_infer.py", line 463, in main
val(input_args)
File "/root/hanyi/Yolov8_for_PyTorch/ultralytics/../om_infer.py", line 433, in val
preds = om_model.infer([batch['img']])
File "/root/hanyi/Yolov8_for_PyTorch/venv/lib/python3.9/site-packages/ais_bench/infer/interface.py", line 162, in infer
return self.run(inputs, out_array=True)
File "/root/hanyi/Yolov8_for_PyTorch/venv/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
请问该如何解决?