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
-
Notifications
You must be signed in to change notification settings - Fork 0
users-9706/Rtdetr-Tensorrt
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published