diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py index 6b06d81..e7cb2d1 100644 --- a/python/tasks/mlb_player.py +++ b/python/tasks/mlb_player.py @@ -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]