facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
https://detectron2.readthedocs.io/en/latest/
Apache License 2.0
30.46k stars 7.48k forks source link

Getting COCO categories in the order in which it is trained #1145

Closed ghost closed 4 years ago

ghost commented 4 years ago

From the model zoo, I am using the Cascade Mask RCNN model pre-trained on MSCOCO. I would like to do model surgery for one of my applications. How can I find out the order of the COCO categories by which the pre-trained model was trained on MSCOCO? For instance, MaskRCNN-benchmark gives the order of the categories by which they trained Mask RCNN model here.

ppwwyyxx commented 4 years ago

You can print the "thing_classes" metadata of a dataset using the example at https://detectron2.readthedocs.io/tutorials/datasets.html#metadata-for-datasets