Some changes to allow training with kaggle data

2023-04-13 14:55:16 +00:00
parent c49645d7bc
commit 7c973f1b88
13 changed files with 1933 additions and 102 deletions
--- a/preprocessing/split_dataset.py
+++ b/preprocessing/split_dataset.py
@@ -8,7 +8,6 @@ dataset = "data/processed/spoter.csv"

 # read the dataset
 df = pd.read_csv(dataset)
-df = map_blazepose_df(df)

 with open("data/sign_to_prediction_index_map.json", "r") as f:
    sign_to_prediction_index_max = json.load(f)