modified ultralytics src code yolov8、yolo11系列模型，yolov5、yolov7等需要对应修改

vim /usr/local/lib/python3.10/dist-packages/ultralytics/nn/tasks.py

BaseModel._predict_once
in end add:

        if isinstance(x, torch.Tensor):
            # import pdb
            # pdb.set_trace()
            x = x.transpose(2, 1)
        return x

export onnx:

from ultralytics import YOLO
model = YOLO("${path}/yolo11n.pt")
success = model.export(format="onnx", simplify=True)  # export the model to onnx format

build engine:

trtexec --onnx=${path}/yolo11n.onnx --saveEngine=${path}/yolo11n.engine

start run

make dir build && cd build
cmake ..
make
./yolo11_det -d ${path}/yolo11n.engine ../images g

部分代码参考wang-xinyu，特此提出感谢

*https://github.com/wang-xinyu/tensorrtx