急急急!客户用官方mindspore的yolov5代码仓源码训练自己的业务数据,训练好的ckpt导出onnx,在atlas200 dk a2上用atc转成om,然后用om替换atlas200 dk里的yolo示例里的om,推理报错
收藏回复举报
急急急!客户用官方mindspore的yolov5代码仓源码训练自己的业务数据,训练好的ckpt导出onnx,在atlas200 dk a2上用atc转成om,然后用om替换atlas200 dk里的yolo示例里的om,推理报错
t('forum.solved') 已解决
发表于2023-12-08 11:23:08
0 查看

客户用官方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

我要发帖子