YOLOV5 TOP-K inference

Viewed 21

Is there a way to get the top 5 categories per bounding box??

I am trying to implement a feature using YOLOv5 that will allow me to output the top 5 possible classifications of a given bounding box in an image and provide them to a user to choose from. I was wondering if there was a way to do this using YOLO as a black box? Or if not, and I have to change something about the code, what that change would have to be?

Thanks!

0 Answers
Related