-
Notifications
You must be signed in to change notification settings - Fork 6
/
nyuName2ID.py
53 lines (50 loc) · 1.26 KB
/
nyuName2ID.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
import numpy as np
SCANNET_COLORMAP = [
[0., 0., 0.],
[174., 199., 232.],
[152., 223., 138.],
[31., 119., 180.],
[255., 187., 120.],
[188., 189., 34.],
[140., 86., 75.],
[255., 152., 150.],
[214., 39., 40.],
[197., 176., 213.],
[148., 103., 189.],
[196., 156., 148.],
[23., 190., 207.],
[247., 182., 210.],
[66., 188., 102.],
[219., 219., 141.],
[140., 57., 197.],
[202., 185., 52.],
[51., 176., 203.],
[200., 54., 131.],
[92., 193., 61.],
[78., 71., 183.],
[172., 114., 82.],
[255., 127., 14.],
[91., 163., 138.],
[153., 98., 156.],
[140., 153., 101.],
[158., 218., 229.],
[100., 125., 154.],
[178., 127., 135.],
[146., 111., 194.],
[44., 160., 44.],
[112., 128., 144.],
[96., 207., 209.],
[227., 119., 194.],
[213., 92., 176.],
[94., 106., 211.],
[82., 84., 163.],
[100., 85., 144.]]
SCANNET_COLORMAP = np.asarray(SCANNET_COLORMAP) / 255.
class_names = []
nyuName2ID = {}
for i, line in enumerate(open("monte_carlo_model_search/label_names.txt").readlines()):
class_id = i # starts with -1
class_name = line.strip()
nyuName2ID[class_name] = class_id
class_names.append(class_name)
class_names = tuple(class_names)