lokesh341 commited on
Commit
ec20eec
·
verified ·
1 Parent(s): ddc7c61

Update services/video_service.py

Browse files
Files changed (1) hide show
  1. services/video_service.py +3 -2
services/video_service.py CHANGED
@@ -6,8 +6,9 @@ VIDEO_DIR = "data"
6
  video_files = [os.path.join(VIDEO_DIR, file) for file in sorted(os.listdir(VIDEO_DIR)) if file.endswith((".mp4", ".avi"))]
7
  video_index = 0
8
  cap = None
 
9
 
10
- def get_next_video_frame(frame_skip=1):
11
  global cap, video_index
12
 
13
  if not video_files:
@@ -16,7 +17,7 @@ def get_next_video_frame(frame_skip=1):
16
  if cap is None or not cap.isOpened():
17
  cap = cv2.VideoCapture(video_files[video_index])
18
 
19
- for _ in range(frame_skip):
20
  ret, frame = cap.read()
21
  if not ret:
22
  cap.release()
 
6
  video_files = [os.path.join(VIDEO_DIR, file) for file in sorted(os.listdir(VIDEO_DIR)) if file.endswith((".mp4", ".avi"))]
7
  video_index = 0
8
  cap = None
9
+ FRAME_SKIP = 2 # Skip every 2 frames for faster processing
10
 
11
+ def get_next_video_frame():
12
  global cap, video_index
13
 
14
  if not video_files:
 
17
  if cap is None or not cap.isOpened():
18
  cap = cv2.VideoCapture(video_files[video_index])
19
 
20
+ for _ in range(FRAME_SKIP):
21
  ret, frame = cap.read()
22
  if not ret:
23
  cap.release()