I am sorry, i think i am just an idiot if i follow the tutorial from TorchVision Object Detection Finetuning Tutorial — PyTorch Tutorials 1.10.0+cu102 documentation 1 where do i have to make changes to add more classes for the mask rcnn model. I have a dataset containing png masks and trying to segment two classes 1. legs, 2. A4 paper. I am spinning in circle for few days to get a more straight forward answer and i can’t really find anything…
P.S the COLAB version from top of the page is not working