How to make Detectron2 demo work on combination of vgg, ImageNet, and COCO/COCO Stuff class names?

Currently, this demo only works for 1600 VGG class names. How can I make it work for union of VGG, COCO, COCO stuff, and ImageNet class names?

Additionally, if say I have 700 other classes that are not in the union of these 4, what would be some straighforward ways to go after it and be able to use in this demo?

My best guess is that you would have to add the class names to the MetadataCatalog.

1 Like