import cv2 | |
from config import FRAME_SKIP | |
def extract_frames(video_path): | |
cap = cv2.VideoCapture(video_path) | |
frames = [] | |
i = 0 | |
while True: | |
ret, frame = cap.read() | |
if not ret: | |
break | |
if i % FRAME_SKIP == 0: | |
frames.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)) | |
i += 1 | |
cap.release() | |
return frames | |