Please provide the following information when requesting support.
• Hardware (x86_ Linux ubuntu jammy 22.04, RTX 3090)
• Network Type (Mask2Former)
• Training spec file(
spec.txt (1.6 KB)
)
I have trained my .pth mask2former model on my custom dataset of 5 classes . But i need to run inference of the trained model with an input image and get the prediction mask value of it .These operation like loading model and doing prediction should be in a .py file app, I need to know how to do it
10 posts - 2 participants