YOLOv8-TensorRT icon indicating copy to clipboard operation
YOLOv8-TensorRT copied to clipboard

Published 20 hours ago • verified publisher icon triple-Mu

Yolov8 Classification

Open advait2000 opened this issue 1 year ago • 3 comments

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

advait2000 avatar Feb 23 '24 14:02 advait2000

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

triple-Mu avatar Mar 01 '24 15:03 triple-Mu

yolov8-cls +1

monkeycc avatar Apr 18 '24 16:04 monkeycc

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

triple-Mu avatar Apr 27 '24 14:04 triple-Mu