| 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 | |