-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathconstants.py
30 lines (24 loc) · 1.23 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
# depth map size : [height, width]
DEPTH_SIZE = [424, 512]
# number of joints
JOINT_COUNT = 25
# maximum number of people
BODY_COUNT = 6
# connecting joint id
CONNECTING_JOINTS = [1, 0, 20, 2, 20, 4, 5, 6, 20, 8, 9, 10, 0, 12, 13, 14, 0, 16, 17, 18, 1, 7, 7, 11, 11]
SLICED_CONNECTING_JOINTS = [
[15, 14, 13, 12, 0], #오른쪽다리
[19, 18, 17, 16, 0], #왼쪽다리
[0, 1, 20, 2, 3], #몸통
[6, 5, 4, 20, 8, 9, 10], #양팔
[23, 11, 10, 24], #왼손
[21, 7, 6, 22]] #오른손
# colors to draw bodies
BODY_COLORS = [(204, 0, 0), (0, 204, 0), (0, 0, 204), (0, 204, 204), (204, 0, 204), (204, 204, 0)]
COLORS = ['r', 'g', 'b', 'c', 'm', 'y']
LINE_COLORS = [tuple(int(num / 2) for num in item) for item in BODY_COLORS]
# colors to draw joints
JOINT_COLORS = [(0, 0, 160), (64, 0, 128), (255, 128, 64), (64, 128, 128), (255, 128, 192), (0, 255, 0),
(128, 64, 64), (0, 128, 255), (128, 128, 128), (128, 128, 0), (0, 255, 255), (255, 128, 64),
(64, 0, 128), (128, 64, 0), (0, 0, 0), (128, 128, 192), (0, 64, 128), (64, 0, 64), (128, 0, 255),
(255, 0, 255), (0, 128, 192), (0, 128, 64), (0, 0, 64), (255, 255, 128), (0, 128, 128)]