-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconstants.py
109 lines (109 loc) · 1.88 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
LABELS_DATA_PATH = "./data/labels.csv"
TASK3_CLASS_COL = "standard"
TASK2_CLASS_LABELS = ['bathroom', 'bedroom', 'dining_room', 'house', 'kitchen', 'living_room']
ROOM_LABELS = ["Bathroom", "Dining room", "Bedroom", "Living room", "Kitchen", "House"]
OBJECT_LABELS = [
"Bathroom cabinet",
"Bathroom sink",
"Bathtub",
"Bed",
"Bed frame",
"Bed sheet",
"Cabinetry",
"Ceiling",
"Chair",
"Chandelier",
"Chest of drawers",
"Coffee table",
"Couch",
"Countertop",
"Cupboard",
"Curtain",
"Dining room",
"Door",
"Drawer",
"Facade",
"Fireplace",
"Floor",
"Furniture",
"Grass",
"Hardwood",
"Kitchen & dining room table",
"Kitchen stove",
"Mattress",
"Nightstand",
"Plumbing fixture",
"Property",
"Real estate",
"Refrigerator",
"Roof",
"Room",
"Rural area",
"Shower",
"Sink",
"Sky",
"Table",
"Tablecloth",
"Tap",
"Tile",
"Toilet",
"Tree",
"Urban area",
"Wall",
"Window",
]
ALL_COLS = [
"Bathroom",
"Bathroom cabinet",
"Bathroom sink",
"Bathtub",
"Bed",
"Bed frame",
"Bed sheet",
"Bedroom",
"Cabinetry",
"Ceiling",
"Chair",
"Chandelier",
"Chest of drawers",
"Coffee table",
"Couch",
"Countertop",
"Cupboard",
"Curtain",
"Dining room",
"Door",
"Drawer",
"Facade",
"Fireplace",
"Floor",
"Furniture",
"Grass",
"Hardwood",
"House",
"Kitchen",
"Kitchen & dining room table",
"Kitchen stove",
"Living room",
"Mattress",
"Nightstand",
"Plumbing fixture",
"Property",
"Real estate",
"Refrigerator",
"Roof",
"Room",
"Rural area",
"Shower",
"Sink",
"Sky",
"Table",
"Tablecloth",
"Tap",
"Tile",
"Toilet",
"Tree",
"Urban area",
"Wall",
"Window",
]