当前位置:   article > 正文

yolov8:pt 转 onnx_yolov8 .pt转onnx

yolov8 .pt转onnx

yolov8官方教程

1.安装包


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

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

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

2.进行模型转换

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

  1. from ultralytics import YOLO
  2. model = YOLO("best.pt")
  3. model.export(format = "onnx") # export the model to onnx format

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号