/
labels.py
31 lines (20 loc) · 910 Bytes
/
labels.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
#!/usr/bin/python
#
# KITTI and MOT labels
#
from collections import namedtuple
#--------------------------------------------------------------------------------
# Definitions
#--------------------------------------------------------------------------------
# label and all information
Label = namedtuple('Label',['name','Id','trainId','color'])
#--------------------------------------------------------------------------------
# A list of all labels
#--------------------------------------------------------------------------------
kitti_labels = [
# name id trainId color
Label( 'unlabeled' , 0 , 10 , (255, 255, 255) ),
Label( 'background' , 0 , 0 , (255, 255, 255) ),
Label( 'person' , 24 , 2 , (220, 20, 60) ),
Label( 'car' , 26 , 1 , ( 0, 0, 142) ),
]