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?

  
      github.com
  
  
    airsplay/py-bottom-up-attention/blob/master/demo/demo_vg_detection.ipynb 3
{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "# You may need to restart your runtime prior to this, to let your installation take effect\n",
    "# Some basic setup\n",
    "# Setup detectron2 logger\n",
    "import detectron2\n",
    "from detectron2.utils.logger import setup_logger\n",
    "setup_logger()\n",
    "import os\n",
    "\n",
    "# import some common libraries\n",
    "import numpy as np\n",
    "import cv2\n",
    "import torch\n",


  This file has been truncated. show original

  
  
    
    
  
  


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?