yolo-tensorrt icon indicating copy to clipboard operation
yolo-tensorrt copied to clipboard

Published 20 hours ago • verified publisher icon enazoe

yolo7 tiny?

Open tugbakara opened this issue 1 year ago • 3 comments

Is this package usable for yolo7 tiny?

tugbakara avatar May 25 '23 14:05 tugbakara

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.

Nuzhny007 avatar May 26 '23 10:05 Nuzhny007

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!

tugbakara avatar May 26 '23 11:05 tugbakara

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.

Nuzhny007 avatar May 27 '23 04:05 Nuzhny007