search

enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.
MIT License
1.18k stars 316 forks source link

yolo7 tiny? #193

Open tugbakara opened 1 year ago

tugbakara commented 1 year ago

Is this package usable for yolo7 tiny?

Nuzhny007 commented 1 year ago

Really, no. This project used a darknet-style config and it need to manually support for all versions of new YOLOs. I think that a simplier way - export a pt model to onnx, converting it to TensorRT and using. In my project I'm going to this way.

tugbakara commented 1 year ago

Actually I am a bit away from this, looking around easy to use things, so in that case could you explain or give a link for the part that you said " - export a pt model to onnx, converting it to TensorRT" I need to learn how can I do that ? Thanks!

Nuzhny007 commented 1 year ago

You can try this complete example: https://github.com/Linaom1214/TensorRT-For-YOLO-Series/tree/main

I'm using the same approach in my project: https://github.com/Smorodov/Multitarget-tracker/blob/master/src/Detector/tensorrt_yolo/YoloONNX.cpp But it is the part of a big project.