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
小王 阅读(12)
评论(0) 编辑 收藏 引用 所属分类:
CUDA