Skip to content

Latest commit

 

History

History
7 lines (7 loc) · 312 Bytes

README.md

File metadata and controls

7 lines (7 loc) · 312 Bytes
Environment: win11, pytorch2.0, cuda11.8, cudnn8.6, tensorrt8.6, Opencv4.8
1. Export onnx
    from ultralytics import YOLO 
    model = YOLO("rtdetr-l.pt") 
    model.export(format="onnx", opset=16)
2. Generate engine
    trtexec.exe --onnx=rtdetr-l.onnx --saveEngine=rtdetr-l.engine