import csv import os import numpy as np

def dump_for_the_app(): pose_samples_folder = 'squat_csvs_out' pose_samples_csv_path = 'squat_csvs_out_basic.csv' file_extension = 'csv' file_separator = ','

Each file in the folder represents one pose class.

file_names = [name for name in os.listdir(pose_samples_folder) if name.endswith(file_extension)]

with open(pose_samples_csv_path, 'w') as csv_out: csv_out_writer = csv.writer(csv_out, delimiter=file_separator, quoting=csv.QUOTE_MINIMAL) for file_name in file_names: # Use file name as pose class name. class_name = file_name[:-(len(file_extension) + 1)]

  # One file line: 'sample_00001,x1,y1,x2,y2,....'.
  with open(os.path.join(pose_samples_folder, file_name)) as csv_in:
    csv_in_reader = csv.reader(csv_in, delimiter=file_separator)
    for row in csv_in_reader:
      row.insert(1, class_name)
      csv_out_writer.writerow(row)

print(f'Pose samples dumped to {pose_samples_csv_path}')

dump_for_the_app()