Fix background averaging. Close #11

colinlaney · Jun 14, 2021 · 9f9fa58 · 9f9fa58
1 parent 31e178e
commit 9f9fa58
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/track.py b/track.py
@@ -162,7 +162,8 @@ def trace(filename):
         ret, frame = cap.read()
         if frame is None:
             break
-        background = cv2.addWeighted(frame, 0.5 - i_frame / n_frames, background, 0.5 + i_frame / n_frames, 0)
+        background = cv2.addWeighted(frame, 0.5 * (1 - i_frame / n_frames),
+                                background, 0.5 * (1 + i_frame / n_frames), 0)
         i_frame += 1
     cap = cv2.VideoCapture(filename)
     ret, frame = cap.read()