-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathcommon.py
44 lines (40 loc) · 904 Bytes
/
common.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
import numpy as np
CONVERTER = np.uint8([
0, # unlabeled
0, # building
0, # fence
0, # other
1, # ped
0, # pole
2, # road line
3, # road
4, # sidewalk
0, # vegetation
5, # car
0, # wall
0, # traffic sign
0, # sky
0, # ground
0, # bridge
0, # railtrack
0, # guardrail
0, # trafficlight
0, # static
0, # dynamic
0, # water
0, # terrain
6,
7,
8,
])
COLOR = np.uint8([
( 0, 0, 0), # unlabeled
(220, 20, 60), # ped
(157, 234, 50), # road line
(128, 64, 128), # road
(244, 35, 232), # sidewalk
( 0, 0, 142), # car
(255, 0, 0), # red light
(255, 255, 0), # yellow light
( 0, 255, 0), # green light
])