weiaicunzai/pytorch-cifar100

Draw box on prediction

khawar-islam opened this issue · 1 comments

khawar-islam commented

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 commented

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