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}