search

triple-Mu / YOLOv8-TensorRT

YOLOv8 using TensorRT accelerate !
MIT License
1.23k stars 210 forks source link

Yolov8 Classification #196

Closed advait2000 closed 1 month ago

advait2000 commented 4 months ago

Is there any way to convert yolov8 classification models to engine format and infer

triple-Mu commented 4 months ago

Yolov8 classification's postprocess is just an argmax op. Maybe it is easy for you.

monkeycc commented 3 months ago

yolov8-cls +1

triple-Mu commented 2 months ago

support now https://github.com/triple-Mu/YOLOv8-TensorRT/commit/a9236fc1599fabe5aa09a1aa2b31a7f6b819f92c