search

jkjung-avt / tensorrt_demos

TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet
https://jkjung-avt.github.io/
MIT License
1.74k stars 545 forks source link

Convert onnx-tensorRT yolov5s #543

Closed dFuong closed 2 years ago

dFuong commented 2 years ago

I need to convert yolov5s model with image-size=640x640

Someone can help me?

jkjung-avt commented 2 years ago

Are you trying to convert a DarkNet model or a PyTorch model or else? The code ("yolo_to_onnx.py") in this repo only works for DarkNet models.

dFuong commented 2 years ago

thanks your response, I understand. I'll try.