Hi,
Saw your code and tried to use this as module.
It did work great but im using yolov8 model and the output im getting only contains bboxes and confidences score. I'm not able to get predicted class id from it.
could you please help?
Thanks and good work.