[Feature Request] Confidence Scores and Names
Opened this issue · 2 comments
robertokcanale commented
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 commented
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
lukazso commented
Thanks for your contribution, I'll check your branch asap!