tools / hoho /color_mappings.py
jacklangerman's picture
add more vis
724cc25
import numpy as np
gestalt_color_mapping = {
"unclassified": (215, 62, 138),
"apex": (235, 88, 48),
"eave_end_point": (248, 130, 228),
"flashing_end_point": (71, 11, 161),
"ridge": (214, 251, 248),
"rake": (13, 94, 47),
"eave": (54, 243, 63),
"post": (187, 123, 236),
"ground_line": (136, 206, 14),
"flashing": (162, 162, 32),
"step_flashing": (169, 255, 219),
"hip": (8, 89, 52),
"valley": (85, 27, 65),
"roof": (215, 232, 179),
"door": (110, 52, 23),
"garage": (50, 233, 171),
"window": (230, 249, 40),
"shutter": (122, 4, 233),
"fascia": (95, 230, 240),
"soffit": (2, 102, 197),
"horizontal_siding": (131, 88, 59),
"vertical_siding": (110, 187, 198),
"brick": (171, 252, 7),
"concrete": (32, 47, 246),
"other_wall": (112, 61, 240),
"trim": (151, 206, 58),
"unknown": (127, 127, 127),
"transition_line": (0,0,0),
}
ade20k_color_mapping = {
'wall': (120, 120, 120),
'building;edifice': (180, 120, 120),
'sky': (6, 230, 230),
'floor;flooring': (80, 50, 50),
'tree': (4, 200, 3),
'ceiling': (120, 120, 80),
'road;route': (140, 140, 140),
'bed': (204, 5, 255),
'windowpane;window': (230, 230, 230),
'grass': (4, 250, 7),
'cabinet': (224, 5, 255),
'sidewalk;pavement': (235, 255, 7),
'person;individual;someone;somebody;mortal;soul': (150, 5, 61),
'earth;ground': (120, 120, 70),
'door;double;door': (8, 255, 51),
'table': (255, 6, 82),
'mountain;mount': (143, 255, 140),
'plant;flora;plant;life': (204, 255, 4),
'curtain;drape;drapery;mantle;pall': (255, 51, 7),
'chair': (204, 70, 3),
'car;auto;automobile;machine;motorcar': (0, 102, 200),
'water': (61, 230, 250),
'painting;picture': (255, 6, 51),
'sofa;couch;lounge': (11, 102, 255),
'shelf': (255, 7, 71),
'house': (255, 9, 224),
'sea': (9, 7, 230),
'mirror': (220, 220, 220),
'rug;carpet;carpeting': (255, 9, 92),
'field': (112, 9, 255),
'armchair': (8, 255, 214),
'seat': (7, 255, 224),
'fence;fencing': (255, 184, 6),
'desk': (10, 255, 71),
'rock;stone': (255, 41, 10),
'wardrobe;closet;press': (7, 255, 255),
'lamp': (224, 255, 8),
'bathtub;bathing;tub;bath;tub': (102, 8, 255),
'railing;rail': (255, 61, 6),
'cushion': (255, 194, 7),
'base;pedestal;stand': (255, 122, 8),
'box': (0, 255, 20),
'column;pillar': (255, 8, 41),
'signboard;sign': (255, 5, 153),
'chest;of;drawers;chest;bureau;dresser': (6, 51, 255),
'counter': (235, 12, 255),
'sand': (160, 150, 20),
'sink': (0, 163, 255),
'skyscraper': (140, 140, 140),
'fireplace;hearth;open;fireplace': (250, 10, 15),
'refrigerator;icebox': (20, 255, 0),
'grandstand;covered;stand': (31, 255, 0),
'path': (255, 31, 0),
'stairs;steps': (255, 224, 0),
'runway': (153, 255, 0),
'case;display;case;showcase;vitrine': (0, 0, 255),
'pool;table;billiard;table;snooker;table': (255, 71, 0),
'pillow': (0, 235, 255),
'screen;door;screen': (0, 173, 255),
'stairway;staircase': (31, 0, 255),
'river': (11, 200, 200),
'bridge;span': (255 ,82, 0),
'bookcase': (0, 255, 245),
'blind;screen': (0, 61, 255),
'coffee;table;cocktail;table': (0, 255, 112),
'toilet;can;commode;crapper;pot;potty;stool;throne': (0, 255, 133),
'flower': (255, 0, 0),
'book': (255, 163, 0),
'hill': (255, 102, 0),
'bench': (194, 255, 0),
'countertop': (0, 143, 255),
'stove;kitchen;stove;range;kitchen;range;cooking;stove': (51, 255, 0),
'palm;palm;tree': (0, 82, 255),
'kitchen;island': (0, 255, 41),
'computer;computing;machine;computing;device;data;processor;electronic;computer;information;processing;system': (0, 255, 173),
'swivel;chair': (10, 0, 255),
'boat': (173, 255, 0),
'bar': (0, 255, 153),
'arcade;machine': (255, 92, 0),
'hovel;hut;hutch;shack;shanty': (255, 0, 255),
'bus;autobus;coach;charabanc;double-decker;jitney;motorbus;motorcoach;omnibus;passenger;vehicle': (255, 0, 245),
'towel': (255, 0, 102),
'light;light;source': (255, 173, 0),
'truck;motortruck': (255, 0, 20),
'tower': (255, 184, 184),
'chandelier;pendant;pendent': (0, 31, 255),
'awning;sunshade;sunblind': (0, 255, 61),
'streetlight;street;lamp': (0, 71, 255),
'booth;cubicle;stall;kiosk': (255, 0, 204),
'television;television;receiver;television;set;tv;tv;set;idiot;box;boob;tube;telly;goggle;box': (0, 255, 194),
'airplane;aeroplane;plane': (0, 255, 82),
'dirt;track': (0, 10, 255),
'apparel;wearing;apparel;dress;clothes': (0, 112, 255),
'pole': (51, 0, 255),
'land;ground;soil': (0, 194, 255),
'bannister;banister;balustrade;balusters;handrail': (0, 122, 255),
'escalator;moving;staircase;moving;stairway': (0, 255, 163),
'ottoman;pouf;pouffe;puff;hassock': (255, 153, 0),
'bottle': (0, 255, 10),
'buffet;counter;sideboard': (255, 112, 0),
'poster;posting;placard;notice;bill;card': (143, 255, 0),
'stage': (82, 0, 255),
'van': (163, 255, 0),
'ship': (255, 235, 0),
'fountain': (8, 184, 170),
'conveyer;belt;conveyor;belt;conveyer;conveyor;transporter': (133, 0, 255),
'canopy': (0, 255, 92),
'washer;automatic;washer;washing;machine': (184, 0, 255),
'plaything;toy': (255, 0, 31),
'swimming;pool;swimming;bath;natatorium': (0, 184, 255),
'stool': (0, 214, 255),
'barrel;cask': (255, 0, 112),
'basket;handbasket': (92, 255, 0),
'waterfall;falls': (0, 224, 255),
'tent;collapsible;shelter': (112, 224, 255),
'bag': (70, 184, 160),
'minibike;motorbike': (163, 0, 255),
'cradle': (153, 0, 255),
'oven': (71, 255, 0),
'ball': (255, 0, 163),
'food;solid;food': (255, 204, 0),
'step;stair': (255, 0, 143),
'tank;storage;tank': (0, 255, 235),
'trade;name;brand;name;brand;marque': (133, 255, 0),
'microwave;microwave;oven': (255, 0, 235),
'pot;flowerpot': (245, 0, 255),
'animal;animate;being;beast;brute;creature;fauna': (255, 0, 122),
'bicycle;bike;wheel;cycle': (255, 245, 0),
'lake': (10, 190, 212),
'dishwasher;dish;washer;dishwashing;machine': (214, 255, 0),
'screen;silver;screen;projection;screen': (0, 204, 255),
'blanket;cover': (20, 0, 255),
'sculpture': (255, 255, 0),
'hood;exhaust;hood': (0, 153, 255),
'sconce': (0, 41, 255),
'vase': (0, 255, 204),
'traffic;light;traffic;signal;stoplight': (41, 0, 255),
'tray': (41, 255, 0),
'ashcan;trash;can;garbage;can;wastebin;ash;bin;ash-bin;ashbin;dustbin;trash;barrel;trash;bin': (173, 0, 255),
'fan': (0, 245, 255),
'pier;wharf;wharfage;dock': (71, 0, 255),
'crt;screen': (122, 0, 255),
'plate': (0, 255, 184),
'monitor;monitoring;device': (0, 92, 255),
'bulletin;board;notice;board': (184, 255, 0),
'shower': (0, 133, 255),
'radiator': (255, 214, 0),
'glass;drinking;glass': (25, 194, 194),
'clock': (102, 255, 0),
'flag': (92, 0, 255),
}
# edge_colors = np.asarray([(214, 251, 248),
# (13, 94, 47),
# (54, 243, 63),
# (187, 123, 236),
# (162, 162, 32),
# (169, 255, 219),
# (8, 89, 52),
# (85, 27, 65),
# (0, 0, 0)]
# edge_colors = np.array([[ 54, 243, 63],
# [214, 251, 248],
# [169, 255, 219],
# [ 13, 94, 47],
# [162, 162, 32],
# [187, 123, 236],
# [ 85, 27, 65],
# [ 0, 0, 0]])