Пример #1
0
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):
Пример #2
0
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)