from scipy.cluster.hierarchy import linkage, dendrogram from scipy.cluster.hierarchy import fcluster from sklearn.metrics import silhouette_score, davies_bouldin_score import glob from tqdm import tqdm import re #from BodyColumn import Holding_joint as HJ from BodyColumn import Upper_joint as UJ motionlist = glob.glob("/home/kei/document/experiments/Master/*.csv") #namelist = [2,3,6,7,9,11,14] namelist = [] motion_num = 0 for motion in tqdm(motionlist): dfpose = pd.read_csv(motion) unidf = rs.dfSpline(dfpose, 500) motionpass_list = motion.split(".") subject_list = motion.split("/") name_component = re.split("[./]", motion) unidf.to_csv("/home/kei/document/experiments/Master/Unified/" + subject_list[-1], index=0) motion_num += 1 namelist.append(name_component[-2]) #選択した関節の数によって行列の数が変わるので,それに合わせて変更する OpenPoseJoint, bodycolumns, Dis_Mat_list = UJ.Member(motion_num) Unified_motion_list = glob.glob( "/home/kei/document/experiments/Master/Unified/*.csv") print(Unified_motion_list) col = 0 for Unified_motion in tqdm(Unified_motion_list):
import pandas as pd import numpy as np import sys import matplotlib.pyplot as plt from procrustes import Procrustes from uniframe import resampling csvpass1 = "/home/kei/document/experiments/ICTH2019/SY/SY2.csv" csvpass2 = "/home/kei/document/experiments/ICTH2019/SK/SK1.csv" df1 = pd.read_csv(csvpass1) df2 = pd.read_csv(csvpass2) uni_df1 = resampling.dfSpline(df1, 1000) uni_df2 = resampling.dfSpline(df2, 1000) uni_df1.to_csv("/home/kei/document/experiments/ICTH2019/SY/UniSY2.csv", index=0) uni_df2.to_csv("/home/kei/document/experiments/ICTH2019/SK/UniSK1.csv", index=0) df1 = pd.read_csv("/home/kei/document/experiments/ICTH2019/SY/UniSY2.csv", index_col=0) df2 = pd.read_csv("/home/kei/document/experiments/ICTH2019/SK/UniSK1.csv", index_col=0) df1_Mat = df1.values df2_Mat = df2.values [d, Z, t] = Procrustes.procrustes(df1_Mat, df2_Mat) #print(Z.shape) rotated_df2 = pd.DataFrame(data=Z, columns=df1.columns) rotated_df2.to_csv( "/home/kei/document/experiments/ICTH2019/SY/Rotated2SK1SY1.csv", index=0)