Yolov8部署推理出错
收藏回复举报
Yolov8部署推理出错
t('forum.solved') 已解决
新人帖
发表于2024-01-11 10:57:14
0 查看
您好,我在参考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板端部署,目前转模型已经成功,但在执行
python3 om_infer.py --weight=yolov8n.pt --om=yolov8n_bs8.om --device_id=0

这条语句的时候报错:[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 

请问该如何解决?

我要发帖子