1.安装yolov8
# Install the required package for YOLOv8 pip install ultralytics
2.模型转换
from ultralytics import YOLO
# Load the YOLOv8 model
model = YOLO("yolov8n.pt")
# Export the model to ONNX format
model.export(format="onnx") # creates 'yolov8n.onnx'
# Load the exported ONNX model
onnx_model = YOLO("yolov8n.onnx")
# Run inference
result = onnx_model.predict(source='1.png', save=True)
print(result)
简化
from ultralytics import YOLO
# Load a YOLOv8 model
model = YOLO("device.pt")
# Export the model
model.export(format="onnx", opset=12, simplify=True, dynamic=False, imgsz=640)
参考: https://blog.csdn.net/qq_29402011/article/details/140937125
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。