Many classes in one image

Hello ,

I have trained my model and have excellent accuracy (as expected!) . I need to load my pth model and predict the classes in images with many objects. What I was doing at the past with tensorflow, was creating rectangulars with the classification on each object .

Can someone helps me how to do it with

Thanks in advance!