3、PyTorch训练YOLOv11—部署篇(detect)—Windows系统

夕西行 / 2024-10-09 / 原文

1、pt转onnx或engine

from ultralytics import YOLO

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

# 预测看下效果
onnx_model = YOLO("best.onnx", task="detect")
results = onnx_model("https://ultralytics.com/images/bus.jpg")
results[0].show()

'''
tensorrt_model = YOLO("best.engine", task="detect")
results = tensorrt_model("https://ultralytics.com/images/bus.jpg")
results[0].show()
'''

2、部署框架的选择

onnxruntime、paddlepaddle等

、、、、未完待续!