How does the augment_train_data function work?

[stonyhu]

@papagina Hello, I am very interested in your work. But I have a question here when reading your code. Could you kindly explain how the augment_train_data function works below?

def get_train_data(bvh_filename):
    
    data=parse_frames(bvh_filename)
    train_data=get_training_format_data(data, non_end_bones,skeleton)
    center=get_motion_center(train_data) #get the avg position of the hip
    center[1]=0.0 #don't center the height

    new_train_data=augment_train_data(train_data, -center, [0,1,0, 0.0])
    return new_train_data