-
Notifications
You must be signed in to change notification settings - Fork 13
/
constants.py
112 lines (104 loc) · 2.36 KB
/
constants.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
coco_categories = {
"chair": 0,
"couch": 1,
"potted plant": 2,
"bed": 3,
"toilet": 4,
"tv": 5,
"dining-table": 6,
"oven": 7,
"sink": 8,
"refrigerator": 9,
"book": 10,
"clock": 11,
"vase": 12,
"cup": 13,
"bottle": 14
}
category_to_id = [
"chair",
"bed",
"plant",
"toilet",
"tv_monitor",
"sofa"
]
category_to_id_gibson = [
"chair",
"couch",
"potted plant",
"bed",
"toilet",
"tv"
]
mp3d_category_id = {
'void': 1,
'chair': 2,
'sofa': 3,
'plant': 4,
'bed': 5,
'toilet': 6,
'tv_monitor': 7,
'table': 8,
'refrigerator': 9,
'sink': 10,
'stairs': 16,
'fireplace': 12
}
# mp_categories_mapping = [4, 11, 15, 12, 19, 23, 6, 7, 15, 38, 40, 28, 29, 8, 17]
mp_categories_mapping = [4, 11, 15, 12, 19, 23, 26, 24, 28, 38, 21, 16, 14, 6, 16]
hm3d_category = [
"chair",
"sofa",
"plant",
"bed",
"toilet",
"tv_monitor",
"bathtub",
"shower",
"fireplace",
"appliances",
"towel",
"sink",
"chest_of_drawers",
"table",
"stairs"
]
coco_categories_mapping = {
56: 0, # chair
57: 1, # couch
58: 2, # potted plant
59: 3, # bed
61: 4, # toilet
62: 5, # tv
60: 6, # dining-table
69: 7, # oven
71: 8, # sink
72: 9, # refrigerator
73: 10, # book
74: 11, # clock
75: 12, # vase
41: 13, # cup
39: 14, # bottle
}
color_palette = [
1.0, 1.0, 1.0,
0.6, 0.6, 0.6,
0.95, 0.95, 0.95,
0.96, 0.36, 0.26,
0.12156862745098039, 0.47058823529411764, 0.7058823529411765,
0.9400000000000001, 0.7818, 0.66,
0.9400000000000001, 0.8868, 0.66,
0.8882000000000001, 0.9400000000000001, 0.66,
0.7832000000000001, 0.9400000000000001, 0.66,
0.6782000000000001, 0.9400000000000001, 0.66,
0.66, 0.9400000000000001, 0.7468000000000001,
0.66, 0.9400000000000001, 0.8518000000000001,
0.66, 0.9232, 0.9400000000000001,
0.66, 0.8182, 0.9400000000000001,
0.66, 0.7132, 0.9400000000000001,
0.7117999999999999, 0.66, 0.9400000000000001,
0.8168, 0.66, 0.9400000000000001,
0.9218, 0.66, 0.9400000000000001,
0.9400000000000001, 0.66, 0.8531999999999998,
0.9400000000000001, 0.66, 0.748199999999999]