JiayuanWang-JW / YOLOv8-multi-task

GNU Affero General Public License v3.0
235 stars 39 forks source link

Tensor RT conversion #63

Open Venkat-1405 opened 3 months ago

Venkat-1405 commented 3 months ago

Hi anyone know how to convert this multi-task model to .onnx (or .trt) and use it by TensorRT? I want to deploy this model in jetson device, thanks

vutuanhungkkk commented 2 months ago

@Venkat-1405 Did you succeed to convert this multi-task model to .onnx (or .trt)? Please help me.

Venkat-1405 commented 2 months ago

@Venkat-1405 Did you succeed to convert this multi-task model to .onnx (or .trt)? Please help me.

I have converted ultralytics pretrained models (yolo v8 seg.pt) to .engine format but couldn't convert Mutitask models (v4s.pt)