24 lines
768 B
Python
24 lines
768 B
Python
import numpy as np
|
|
|
|
|
|
def transform(name, arr: np.ndarray):
|
|
if name == 'HumanAct12':
|
|
rotate = [1., -1., -1.]
|
|
for i in range(arr.shape[0]):
|
|
origin = arr[i][0].copy()
|
|
for j in range(arr.shape[1]):
|
|
arr[i][j] -= origin
|
|
for k in range(3):
|
|
arr[i][j][k] *= rotate[k]
|
|
arr[i][0] = [0.0, 0.0, 0.0]
|
|
elif name == 'UTD_MHAD':
|
|
rotate = [-1., 1.,-1.]
|
|
for i in range(arr.shape[0]):
|
|
origin = arr[i][3].copy()
|
|
for j in range(arr.shape[1]):
|
|
arr[i][j] -= origin
|
|
for k in range(3):
|
|
arr[i][j][k] *= rotate[k]
|
|
arr[i][3] = [0.0, 0.0, 0.0]
|
|
return arr
|