yoga28v28/Video_temporal_error_detection

1
2
3import cv2
4import numpy as np
5
6# ✅ MOTION FUNCTION (MISSING — ADD THIS)
7def compute_motion(prev_frame, curr_frame):
8    prev_gray = cv2.cvtColor(prev_frame, cv2.COLOR_BGR2GRAY)
9    curr_gray = cv2.cvtColor(curr_frame, cv2.COLOR_BGR2GRAY)
10
11    flow = cv2.calcOpticalFlowFarneback(
12        prev_gray, curr_gray, None,
13        0.5, 3, 15, 3, 5, 1.2, 0
14    )
15
16    magnitude = np.sqrt(flow[...,0]**2 + flow[...,1]**2)
17    return np.mean(magnitude)
18
19
20# ✅ MERGE DETECTION (UPDATED)
21def detect_merge(prev_frame, curr_frame):
22    diff = cv2.absdiff(prev_frame, curr_frame)
23    score = diff.mean()
24
25    return score < 0.7, score