Spaces:
Runtime error
Runtime error
File size: 423 Bytes
ec313eb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import cv2
from facefusion.types import VideoPoolSet
VIDEO_POOL_SET : VideoPoolSet = {}
def get_video_capture(video_path : str) -> cv2.VideoCapture:
if video_path not in VIDEO_POOL_SET:
VIDEO_POOL_SET[video_path] = cv2.VideoCapture(video_path)
return VIDEO_POOL_SET.get(video_path)
def clear_video_pool() -> None:
for video_capture in VIDEO_POOL_SET.values():
video_capture.release()
VIDEO_POOL_SET.clear()
|