Spaces:
Build error
Build error
File size: 789 Bytes
a001524 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import os
import tqdm
import pandas as pd
from shutil import copyfile
MAIN_PATH = "/Users/matyasbohacek/Documents/Academics/Projects/AUTSL"
BATCH = "test"
df = pd.read_csv(MAIN_PATH + "/" + BATCH + "_labels.csv", encoding="utf-8", sep=";")
if not os.path.exists(MAIN_PATH + "/" + BATCH + "_preprocessed/"):
os.mkdir(MAIN_PATH + "/" + BATCH + "_preprocessed/")
for index_row, row in tqdm.tqdm(df.iterrows()):
if not os.path.exists(MAIN_PATH + "/" + BATCH + "_preprocessed/" + str(row["label"]) + "/"):
os.mkdir(MAIN_PATH + "/" + BATCH + "_preprocessed/" + str(row["label"]) + "/")
copyfile(MAIN_PATH + "/" + BATCH + "/" + str(row["video"]) + "_color.mp4", MAIN_PATH + "/" + BATCH + "_preprocessed/" + str(row["label"]) + "/" + str(row["video"]) + "_color.mp4")
|