fulyvv/pose2sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improve script

Browse Source
This commit is contained in:
davidpagnon 2023-10-30 14:28:48 +01:00
parent 1c7f4565af
commit da843a6fc8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Pose2Sim/Utilities/trc_Zup_to_Yup.py
View File

@ -63,12 +63,12 @@ def trc_Zup_to_Yup_func(*args):
frames_col, time_col = trc_df.iloc[:,0], trc_df.iloc[:,1] frames_col, time_col = trc_df.iloc[:,0], trc_df.iloc[:,1]
Q_coord = trc_df.drop(trc_df.columns[[0, 1]], axis=1) Q_coord = trc_df.drop(trc_df.columns[[0, 1]], axis=1)
# Y->Z, Z->-Y # Y->Z, Z->Y
cols = list(Q_coord.columns) cols = list(Q_coord.columns)
# cols = np.array([[cols[i*3+1],cols[i*3+2],cols[i*3]] for i in range(int(len(cols)/3))]).flatten() # X->Y, Y->Z, Z->X # cols = np.array([[cols[i*3+1],cols[i*3+2],cols[i*3]] for i in range(int(len(cols)/3))]).flatten() # X->Y, Y->Z, Z->X
cols = np.array([[cols[i*3],cols[i*3+2],cols[i*3+1]] for i in range(int(len(cols)/3))]).flatten() # Y->Z, Z->-Y cols = np.array([[cols[i*3],cols[i*3+2],cols[i*3+1]] for i in range(int(len(cols)/3))]).flatten() # Y->Z, Z->-Y
Q_Yup = Q_coord[cols] Q_Yup = Q_coord[cols]
Q_Yup.iloc[:,2::3] = - Q_Yup.iloc[:,2::3] # Q_Yup.iloc[:,2::3] = - Q_Yup.iloc[:,2::3]
# write file # write file
with open(trc_yup_path, 'w') as trc_o: with open(trc_yup_path, 'w') as trc_o: