yolov8:pt 转 onnx

yolov8官方教程

1.安装包


我使用的是虚拟环境,yolov8包已经下载到本地了,因此直接在anaconda prompt 命令行继续安装

  首先激活自己的虚拟环境,然后执行安装命令

pip install ultralytics
  • yolov8中没有requirement.txt文件,直接安装 ultralytics 即可

yolov8:pt 转 onnx_第1张图片

2.进行模型转换

yolov8:pt 转 onnx_第2张图片

将转换好的pt文件放在weight文件夹下,然后在这个文件夹创建一个py文件,里面使用如下代码进行转换:

from ultralytics import YOLO

model = YOLO("best.pt")
model.export(format = "onnx")  # export the model to onnx format

你可能感兴趣的:(yolo,YOLO)