lukazso/yolov6-ros

[Feature Request] Confidence Scores and Names

Opened this issue · 2 comments

Hi, first of all congrats for the great work! I have tried this and the Yolov7, but the other one unfortunately has memory leak.

Could you please include (and I will be working on it soon):

-Drawing class names and not numbers on the prediction
-Drawing BB prediction confidence score

Actually I have fixed it myself.
I had to add a conf file from COCO to get the classes names, and small changes to the visualizer. Please check the repo I forked from you or I would gladly share the code if you want

Thanks for your contribution, I'll check your branch asap!