yolov8模型转onnx模型和 tensorRT 模型

转onnx模型

在安装好

pip install onnxruntime-gpu

pip install onnx onnxconverter-common

出现 No module named 'cpuinfo' 错误，通过安装：

pip install py-cpuinfo 解决该问题。

import sys
# 即 ultralytics文件夹 所在绝对路径
sys.path.append(r"E:\JupyterNotebook\_02yolov8seg\xiaoJi_yolov8_seg/")from ultralytics import YOLO
#采用源码训练、验证、预测、导出模型,不需要依赖安装ultralytics# Load a model
model = YOLO(r"E:\JupyterNotebook\_02yolov8seg\xiaoJi_yolov8_seg\weights\segexp11best.pt")   # load a pretrained model (recommended for training)
success = model.export(format="onnx")  # export the model to ONNX format  #转换为onnx模型
print('model.export success')

最后导出的onnx模型在你原来的 model 目录下

在导出 onnx模型的时候，还可以加相关的参数，控制导出的onnx的一些性质，参数解释在官网是是解释最权威的，如下：

Python - Ultralytics YOLOv8 Docs

如何加参数，比如加一个 dynamic=True ，参考如下：

model.export(format='onnx', dynamic=True)