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