15 lines
479 B
Python
15 lines
479 B
Python
import numpy as np
|
|
|
|
def mapping(Jtr,cfg):
|
|
name=cfg.DATASET.NAME
|
|
if not name=='HumanAct12':
|
|
mapped_joint=cfg.DATASET.DATA_MAP.UTD_MHAD
|
|
Jtr_mapped=np.zeros([Jtr.shape[0],len(mapped_joint),Jtr.shape[2]])
|
|
for i in range(Jtr.shape[0]):
|
|
for j in range(len(mapped_joint)):
|
|
for k in range(Jtr.shape[2]):
|
|
Jtr_mapped[i][j][k]=Jtr[i][mapped_joint[j]][k]
|
|
return Jtr_mapped
|
|
return Jtr
|
|
|