umaradnaan commited on
Commit
abd72f0
·
verified ·
1 Parent(s): 9a27b3d

Update utils/video_utils.py

Browse files
Files changed (1) hide show
  1. utils/video_utils.py +24 -24
utils/video_utils.py CHANGED
@@ -1,24 +1,24 @@
1
- import cv2
2
- import os
3
-
4
- def extract_frames(video_path, out_dir, every_n=15):
5
- cap = cv2.VideoCapture(video_path)
6
- frames = []
7
- count = 0
8
- idx = 0
9
-
10
- while cap.isOpened():
11
- ret, frame = cap.read()
12
- if not ret:
13
- break
14
-
15
- if count % every_n == 0:
16
- frame_path = os.path.join(out_dir, f"frame_{idx}.jpg")
17
- cv2.imwrite(frame_path, frame)
18
- frames.append(frame_path)
19
- idx += 1
20
-
21
- count += 1
22
-
23
- cap.release()
24
- return frames
 
1
+ import cv2
2
+ import os
3
+
4
+ def extract_frames(video_path, out_dir, every_n=15):
5
+ cap = cv2.VideoCapture(video_path)
6
+ frames = []
7
+ count = 0
8
+ idx = 0
9
+
10
+ while cap.isOpened():
11
+ ret, frame = cap.read()
12
+ if not ret:
13
+ break
14
+
15
+ if count % every_n == 0:
16
+ frame_path = os.path.join(out_dir, f"frame_{idx}.jpg")
17
+ cv2.imwrite(frame_path, frame)
18
+ frames.append(frame_path)
19
+ idx += 1
20
+
21
+ count += 1
22
+
23
+ cap.release()
24
+ return frames