Skip to content

export rect model #158

@enhany

Description

@enhany

After export detect model with rectangle size (example: 320*640), tensorrt engine model doesn't detect anything (with end2end). Without end2end there will be many-many wrong boxes.
Tensorrt: 10.3

code (imgsz added):

from ultralytics import YOLO
model = YOLO("yolo11n.pt")
model.export(format='onnx', imgsz=[320,640])

and then follow manual

python export.py  -o yolo11n.onnx -e yolov11n.trt --end2end --v8 -p fp32
python trt.py -e yolov11n.trt  -i src/1.jpg -o yolov11-1.jpg --end2end 

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions