Labels names of Pretrained Mask RCNN model

vision
1

Hi, everyone,

I have a question about the name of labels predicted by pretrained mask rcnn in Pytorch,
torchvision.models.detection.maskrcnn_resnet50_fpn

I wonder if there is an official name lists indicating the name of each label number, like
0 -->BG
1–>person
…

Thanks

2

Check out this link: Mask_RCNN/demo.ipynb at master · matterport/Mask_RCNN · GitHub

Although the repo is for Keras, I guess it serves your purpose. Class labels remain the same.