Draw box on prediction

[khawar-islam]

Dear @weiaicunzai

I am doing a classification task and I need to draw a box with the class name that the model predicts for the image. Please help me

Regards, Khawar

[MohamedA95]

AFAIK, draw a box is not classification rather object detection {Classification+Localization}