1、在yolov7中把yolov7.pt轉為.onnx
python3 export.py --weights weights/yolov7.pt --grid --simplify --topk-all 100 --img-size 640 640 --max-wh 640
2、在CANN環境下使用act把.onnx轉為.om
atc --model=yolov7.onnx --framework=5 --output=yolov7 --input_shape="images:1,3,640,640" --soc_version=Ascend310 --insert_op_conf=aipp.cfg
3、在C++程序中使用CANN對yolov7.om模型推理，執行aclmdlExecute()函數失敗。
返回錯誤碼：507011
解決辦法：
在阿特拉斯設備中查看設備信息：
npu-smi info
可以看到芯片型號是310B1
修改act命令參數：
atc --model=yolov7.onnx --framework=5 --output=yolov7 --input_shape="images:1,3,640,640" --soc_version=Ascend310B1 --insert_op_conf=aipp.cfg
把生成的.om文件放到C++程序中進行推理，推理成功！！！
O了！
我翹里哇！！！！！！！！！！！！！！！！！！！！！！！！！！
posted on 2024-04-16 20:42
小王 閱讀(366)
評論(0) 編輯 收藏 引用 所屬分類:
昇騰