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