[~] Refactor

This commit is contained in:
Siarhei Siniak 2021-07-27 08:20:38 +03:00
parent 3fe1287bfc
commit 223b239ceb

@ -1335,3 +1335,22 @@ def kernel_19(o_18):
t7,
]
)
def kernel_20(o_18):
l_pair = [
(0, 1), (0, 2), (1, 3), (2, 4), # Head
(5, 6), (5, 7), (7, 9), (6, 8), (8, 10),
(17, 11), (17, 12), # Body
(11, 13), (12, 14), (13, 15), (14, 16)
]
p_color = [(0, 255, 255), (0, 191, 255), (0, 255, 102), (0, 77, 255), (0, 255, 0), # Nose, LEye, REye, LEar, REar
(77, 255, 255), (77, 255, 204), (77, 204, 255), (191, 255, 77), (77, 191, 255), (191, 255, 77), # LShoulder, RShoulder, LElbow, RElbow, LWrist, RWrist
(204, 77, 255), (77, 255, 204), (191, 77, 255), (77, 255, 191), (127, 77, 255), (77, 255, 127), (0, 255, 255)] # LHip, RHip, LKnee, Rknee, LAnkle, RAnkle, Neck
line_color = [(0, 215, 255), (0, 255, 204), (0, 134, 255), (0, 255, 50),
(77, 255, 222), (77, 196, 255), (77, 135, 255), (191, 255, 77), (77, 255, 77),
(77, 222, 255), (255, 156, 127),
(0, 127, 255), (255, 127, 77), (0, 77, 255), (255, 77, 36)]
t1 = o_18['t2']['t7'][0]['keypoints']
t2 = o_18['t2']['t6'][0]
t3 = o_18['t2']['t1'][0]['image_canvas']
assert o_18['t2']['t7'][0]['image_id'] == os.path.split(o_18['t2']['t1'][0])[1]