Spaces:
Sleeping
Sleeping
from typing import Optional | |
import cv2 | |
from DeepFakeAI.typing import Frame | |
def get_video_frame(video_path : str, frame_number : int = 0) -> Optional[Frame]: | |
capture = cv2.VideoCapture(video_path) | |
frame_total = capture.get(cv2.CAP_PROP_FRAME_COUNT) | |
capture.set(cv2.CAP_PROP_POS_FRAMES, min(frame_total, frame_number - 1)) | |
has_frame, frame = capture.read() | |
capture.release() | |
if has_frame: | |
return frame | |
return None | |
def get_video_frame_total(video_path : str) -> int: | |
capture = cv2.VideoCapture(video_path) | |
video_frame_total = int(capture.get(cv2.CAP_PROP_FRAME_COUNT)) | |
capture.release() | |
return video_frame_total | |