Vid2ShapeAI / utils /video_utils.py
umaradnaan's picture
Update utils/video_utils.py
abd72f0 verified
raw
history blame contribute delete
513 Bytes
import cv2
import os
def extract_frames(video_path, out_dir, every_n=15):
cap = cv2.VideoCapture(video_path)
frames = []
count = 0
idx = 0
while cap.isOpened():
ret, frame = cap.read()
if not ret:
break
if count % every_n == 0:
frame_path = os.path.join(out_dir, f"frame_{idx}.jpg")
cv2.imwrite(frame_path, frame)
frames.append(frame_path)
idx += 1
count += 1
cap.release()
return frames