Spaces:
Sleeping
Sleeping
Update services/video_service.py
Browse files
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(
|
| 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(
|
| 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()
|