yolov6-ros icon indicating copy to clipboard operation
yolov6-ros copied to clipboard

[Feature Request] Confidence Scores and Names

Open robertokcanale opened this issue 2 years ago • 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

robertokcanale avatar Jul 15 '22 08:07 robertokcanale

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

robertokcanale avatar Jul 15 '22 09:07 robertokcanale

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

lukazso avatar Jul 18 '22 11:07 lukazso