SathvikGanta commited on
Commit
326dd00
·
verified ·
1 Parent(s): 494c758

Create utils/frame_extractor.py

Browse files
Files changed (1) hide show
  1. utils/frame_extractor.py +15 -0
utils/frame_extractor.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import cv2
2
+
3
+ def extract_frames(video_path, interval=30):
4
+ cap = cv2.VideoCapture(video_path)
5
+ frames = []
6
+ count = 0
7
+ while cap.isOpened():
8
+ ret, frame = cap.read()
9
+ if not ret:
10
+ break
11
+ if count % interval == 0:
12
+ frames.append(frame)
13
+ count += 1
14
+ cap.release()
15
+ return frames