YOLOV8训练好的torch模型转换成ONNX、OM格式

一、pt 转 ONNX 格式

from ultralytics import YOLO model = YOLO("runs/detect/train/weights/best.pt") # load a custom trained model model.export(format="onnx", dynamic=False, simplify=True, opset=11)

注意：export中的参数 opset=11，指定导出的ONNX模型版本版本。版本太高可能会在下一步ONNX转 om 时报错！报错如下：

二、ONNX 模型转 om 格式

前提需要自己下好CANN安装包，安装教程：快速安装CANN-从这里开始-CANN社区版8.0.RC2.alpha003开发文档-昇腾社区 (hiascend.com)https://www.hiascend.com/document/detail/zh/CANNCommunityEdition/80RC2alpha003/quickstart/quickstart/quickstart_18_0005.html

按照以下设置为例子：

atc --model=best.onnx --framework=5 --output=best --input_shape="images:1,3,640,640" --soc_version=Ascend310

--model：参数为onnx模型的路径

--output：参数为导出模型的路径

--soc_version：你的处理器型号

--framework：“5” 表示ONNX格式

出现以下提示表示成功：

ATC run success, welcome to the next use.