当前位置:   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博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/292060
推荐阅读
相关标签
  

闽ICP备14008679号