constant.py

TRAIN_CLASSES = {
    "COCO": ["apple", "backpack", "banana", "bench", "bicycle", "bird", "boat", "book", "bottle", "bowl", "broccoli", "bus", "cake", "car", "carrot", "cell phone", "chair", "clock", "couch", "cow", "cup", "dining table", "dog", "donut", "elephant", "handbag", "horse", "kite", "knife", "motorcycle", "orange", "person", "pizza", "potted plant", "remote", "sheep", "sink", "skateboard", "skis", "spoon", "sports ball", "suitcase", "surfboard", "tie", "traffic light", "truck", "tv", "umbrella", "vase", "wine glass"],
    "ADE": ["arm", "armchair", "bed", "book", "bottle", "box", "building", "cabinet", "car", "ceiling", "chair", "column", "curtain", "cushion", "door", "drawer", "fence", "floors", "flower", "glass", "grass", "handle", "head", "lamp", "leg", "light", "light source", "mirror", "mountain", "pane", "person", "picture", "pillow", "plant", "plate", "pole", "pot", "road", "rock", "seat", "shelf", "sign", "sofa", "spotlight", "streetlight", "table", "tree", "vase", "wheel", "window"]
}

VALIDATION_CLASSES = {
    "COCO": ["dining table", "cow", "spoon", "potted plant", "zebra", "donut", "traffic light", "backpack", "vase", "sports ball", "person", "apple", "chair", "bicycle", "pizza", "remote", "tennis racket", "bench", "boat", "motorcycle", "bottle", "broccoli", "suitcase", "dog", "orange", "skis", "cake", "umbrella", "cell phone", "bowl", "wine glass", "surfboard", "handbag", "sheep", "tie", "banana", "sink", "kite", "clock", "cup", "giraffe", "bus", "knife", "bird", "elephant", "truck", "book", "car", "tv", "carrot"],
    "ADE": ["plant", "window", "glass", "windshield", "vase", "mirror", "tree", "ceiling", "cabinet", "rock", "person", "bag", "chair", "door", "light", "food", "arm", "base", "bottle", "brand", "grass", "box", "pole", "license plate", "curtain", "plate", "mountain", "table", "head", "building", "balcony", "shelf", "pillow", "column", "shutter", "flowerpot", "leg", "apron", "sign", "picture", "cushion", "flower", "drawer", "wheel", "roof", "book", "price tag", "car", "rim", "handle"]
}