SathvikGanta commited on
Commit
9560049
·
verified ·
1 Parent(s): 44883c8

Update utils/video_utils.py

Browse files
Files changed (1) hide show
  1. utils/video_utils.py +3 -7
utils/video_utils.py CHANGED
@@ -1,12 +1,7 @@
1
  import cv2
2
- import tempfile
3
 
4
- def extract_frames(video_file, interval=30):
5
- tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
6
- tmp.write(video_file.read())
7
- tmp.close()
8
-
9
- cap = cv2.VideoCapture(tmp.name)
10
  frames = []
11
  i = 0
12
  while cap.isOpened():
@@ -18,3 +13,4 @@ def extract_frames(video_file, interval=30):
18
  i += 1
19
  cap.release()
20
  return frames
 
 
1
  import cv2
 
2
 
3
+ def extract_frames(video_path, interval=30):
4
+ cap = cv2.VideoCapture(video_path) # directly use the path
 
 
 
 
5
  frames = []
6
  i = 0
7
  while cap.isOpened():
 
13
  i += 1
14
  cap.release()
15
  return frames
16
+