Engine Serialisation Error

[Addoxx]

Hello spacewalk01,

Thanks for sharing your Yolo11 TensorRT implementation!

I am facing this issue where I am not able to convert the weights from PyTorch to an engine file for me to use in this module. Can you please guide me on how to convert the PyTorch file?

Thanks in advance,

[spacewalk01]

First use the following code to convert pytorch model into onnx format. Then you can input your onnx model into the program to generate an engine

from ultralytics import YOLO

# Load the YOLO model
model = YOLO("yolo11s.pt")

# Export the model to ONNX format
export_path = model.export(format="onnx")

print(f"Model exported to {export_path}")