-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathcommon.py
22 lines (19 loc) · 1.41 KB
/
common.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
activity_classes_names = ["Crunches mat", "Aerobic Step", "Uneven bars", "Bar", "Pommel horse",
"elliptical training machine", "Balance Beam", "Rowing machine", "Handle",
"Rope", "Bicycle", "Parallel bars", "Bars"]
coco_classes = ['person', 'bicycle', 'car', 'motorcycle', 'bus',
'traffic light', 'fire hydrant', 'stop sign', 'parking meter', 'bench',
'backpack', 'umbrella', 'frisbee', 'snowboard', 'sports ball',
'baseball bat', 'baseball glove', 'skateboard', 'tennis racket', 'bottle']
all_classes = ['background'] + coco_classes + activity_classes_names
COCO_NUM_CLASSES = len(coco_classes) #80
ACTIVITY_NUM_CLASSES = len(activity_classes_names)
ALL_NUM_CLASSES = len(all_classes)
hex_colors = ['#00FF00','#0000FF','#FF0000','#01FFFE','#FFA6FE','#FFDB66','#006401',
'#010067','#95003A','#007DB5','#FF00F6','#FFEEE8','#774D00','#90FB92','#0076FF',
'#D5FF00','#FF937E','#6A826C','#FF029D','#FE8900','#7A4782','#7E2DD2','#85A900',
'#FF0056','#A42400','#00AE7E','#683D3B','#BDC6FF','#263400','#BDD393','#00B917',
'#9E008E','#001544','#C28C9F','#FF74A3','#01D0FF','#004754','#E56FFE','#788231',
'#0E4CA1','#91D0CB','#BE9970','#968AE8','#BB8800','#43002C','#DEFF74','#00FFC6',
'#FFE502','#620E00','#008F9C','#98FF52','#7544B1','#B500FF','#00FF78','#FF6E41',
'#005F39','#6B6882','#5FAD4E','#A75740','#A5FFD2','#FFB167','#009BFF','#E85EBE']