YOLO(You Only Look Once)
安装YOLOv8
pip install ultralytics -i https://pypi.douban.com/simple/
下载模型权重文件
在官方网站中下载我们所需要的权重文件
注意:这一步其实可以不做,我们在第三步导出模型为onnx的时候如果权重文件不存在,会自动帮我们下载一个权重文件,但速度会较慢,所以个人还是建议先在官网中下载权重文件,再导出为onnx模型
新建一个文件夹名字为“yolov8_onnx”,将刚刚下载的权重文件“yolov8s.pt”放到该文件夹下的models文件夹里
在models文件夹下打开cmd,在cmd中输入以下命令将模型直接导出为onnx模型:
yolo export model=yolov8s.pt format=onnx opset=12
python脚本
from ultralytics import YOLO
# Load a model
model = YOLO("\models\yolov8s.pt") # load an official model
# Export the model
model.export(format="onnx")