constants

TRAIN_SUBJECTS = [1, 5, 6, 7, 8]
TEST_SUBJECTS  = [9, 11]

H36M_NAMES = ['']*32
H36M_NAMES[0]  = 'Hip'
H36M_NAMES[1]  = 'RHip'
H36M_NAMES[2]  = 'RKnee'
H36M_NAMES[3]  = 'RFoot'
H36M_NAMES[6]  = 'LHip'
H36M_NAMES[7]  = 'LKnee'
H36M_NAMES[8]  = 'LFoot'
H36M_NAMES[12] = 'Spine'
H36M_NAMES[13] = 'Thorax'
H36M_NAMES[14] = 'Neck/Nose'
H36M_NAMES[15] = 'Head'
H36M_NAMES[17] = 'LShoulder'
H36M_NAMES[18] = 'LElbow'
H36M_NAMES[19] = 'LWrist'
H36M_NAMES[25] = 'RShoulder'
H36M_NAMES[26] = 'RElbow'
H36M_NAMES[27] = 'RWrist'

N_CAMERAS = 4
N_JOINTS = 32

PLOT_RADIUS = 300

data_path = Path('data')
data_path.ls()

[PosixPath('data/test_2d.pt'),
 PosixPath('data/stat_2d.pt'),
 PosixPath('data/train_3d.pt'),
 PosixPath('data/stat_3d.pt'),
 PosixPath('data/test_2d_ft.pt'),
 PosixPath('data/train_2d_ft.pt'),
 PosixPath('data/test_3d.pt'),
 PosixPath('data/rcams.pt'),
 PosixPath('data/train_2d.pt')]

actions

get_actions[source]

get_actions(action)

assert get_actions('all') == ['Directions', 'Discussion', 'Eating', 'Greeting', 'Phoning', 'Photo', 'Posing', 'Purchases', 'Sitting', 'SittingDown', 'Smoking', 'Waiting', 'WalkDog', 'Walking', 'WalkTogether']
assert get_actions('All') == ['Directions', 'Discussion', 'Eating', 'Greeting', 'Phoning', 'Photo', 'Posing', 'Purchases', 'Sitting', 'SittingDown', 'Smoking', 'Waiting', 'WalkDog', 'Walking', 'WalkTogether']
assert get_actions('Smoking') == ['Smoking']

data_utils

normalize_data[source]

normalize_data(unnormalized, mean, std, dim_use)

unnormalize_data[source]

unnormalize_data(normalized, mean, std, dim_ignore)

normalize_kp[source]

normalize_kp(kp, mean, std, dim_use)

get_kp_from_json[source]

get_kp_from_json(fname)

coco_to_skel[source]

coco_to_skel(s)

cameras

get_cam_rt[source]

get_cam_rt(key, rcams)

camera_to_world_frame[source]

camera_to_world_frame(P, R, T)

cam_to_world_centered[source]

cam_to_world_centered(data, key, rcams)

viz

show_2d_pose[source]

show_2d_pose(skel, ax, lcolor='#094e94', rcolor='#940909')

show_3d_pose[source]

show_3d_pose(skel, ax, lcolor='#094e94', rcolor='#940909')