| import torch | |
| import numpy as np | |
| from scipy.spatial.transform import Rotation as R | |
| def transform_joint_to_other_db(src_joint, src_name, dst_name): | |
| src_joint_num = len(src_name) | |
| dst_joint_num = len(dst_name) | |
| new_joint = np.zeros(((dst_joint_num,) + src_joint.shape[1:]), dtype=np.float32) | |
| for src_idx in range(len(src_name)): | |
| name = src_name[src_idx] | |
| if name in dst_name: | |
| dst_idx = dst_name.index(name) | |
| new_joint[dst_idx] = src_joint[src_idx] | |
| return new_joint |