import pandas as pd import body_dictionary as body_dic body = body_dic.body() dict1 = ['frame', 'x', 'y', 'score', 'body_part'] def matrix_to_csv(dataframe, path, name, dict=dict1): """ Goal: insert dataframe in file csv :param dataframe: :param path: path where it will be insert csv file :param name: name of csv file """ dataframe.to_csv(path + name + '.csv', index=False, columns=dict) def csv_to_matrix(name): """ Goal: read file csv and export dataframe :param name: name of csv file :return: dataframe """ df = pd.read_csv(name) return df # , dict
import numpy as np import pandas as pd from scipy.spatial.distance import cosine from body_dictionary import body bod = body() from csv_tools import * def variance(path, A, B): dist_list = [] for element in path: i = element[0] j = element[1] dist_list.append(calc_var(A[i], B[j])) variance_matrix = np.asarray(dist_list) variance_m = np.max(variance_matrix, axis=0) return variance_m def vec(point_a, point_b): vect = (point_a[0] - point_b[0], point_a[1] - point_b[1], 0) return vect def calc_var(A, B): dist_cos = np.zeros(len(body.connection) + 1) i = -1 for element in body.connection: i = i + 1