Diff of /process_data.py [000000] .. [225570]

Switch to side-by-side view

--- a
+++ b/process_data.py
@@ -0,0 +1,25 @@
+if __name__ == "__main__":
+    import fall_detector
+    import sys
+    import csv
+    import os
+    import joblib
+    import time
+    sub_start = 15
+    sub_end = 18
+    orig_sys_argv = sys.argv
+    for act_id in range(1, 12):
+        for sub_id in range(sub_start, sub_end):
+            dl = []
+            t0 = time.time()
+            for trial_id in range(1, 4):
+                for cam_id in range(1, 3):
+                    if os.path.exists(f'dataset/Activity{act_id}/Subject{sub_id}/Trial{trial_id}Cam1.mp4'):
+                        args = ['--coco_points', f'--video=dataset/Activity{act_id}/Subject{sub_id}/Trial{trial_id}Cam{cam_id}.mp4']
+                        sys.argv = [orig_sys_argv[0]] + args
+                        f = fall_detector.FallDetector()
+                        q1 = f.begin_mixed()
+                        dl.append(q1)
+
+            joblib.dump(dl, f'dataset/Activity{act_id}/Subject{sub_id}/coco.kps', True)
+            print(time.time()-t0)