import numpy as np def prepare_vertices(vertices, canonicalize=True): data = vertices # Swap axis (gravity=Z instead of Y) # data = data[..., [2, 0, 1]] # Make left/right correct # data[..., [1]] = -data[..., [1]] # Center the first root to the first frame data -= data[[0], [0], :] # Remove the floor data[..., 2] -= np.min(data[..., 2]) return data