|
|
import os
|
|
|
import json
|
|
|
from scripts.extract_frames import extract_frames
|
|
|
from scripts.cursor_tracker import track_cursor
|
|
|
|
|
|
def run_pipeline():
|
|
|
video_dir = "videos"
|
|
|
frames_dir = "frames"
|
|
|
cursor_dir = "cursors"
|
|
|
annotations_dir = "annotations"
|
|
|
os.makedirs(frames_dir, exist_ok=True)
|
|
|
os.makedirs(cursor_dir, exist_ok=True)
|
|
|
os.makedirs(annotations_dir, exist_ok=True)
|
|
|
|
|
|
for video_file in os.listdir(video_dir):
|
|
|
if video_file.lower().endswith(('.mp4', '.avi', '.mov')):
|
|
|
extract_frames(os.path.join(video_dir, video_file), frames_dir)
|
|
|
|
|
|
track_cursor(frames_dir, cursor_dir, os.path.join(annotations_dir, "cursor_positions.json"))
|
|
|
print("Pipeline complete.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
run_pipeline()
|
|
|
|
