forked from pkwhiuqat/HumanFallDetectionLSTM
26 lines
974 B
Python
26 lines
974 B
Python
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)
|