mirror of
https://github.com/facefusion/facefusion.git
synced 2026-05-01 22:05:09 +02:00
Henry Ruhs
8801668562
* honor webcam resolution to avoid stripe mismatch, update dependencies * avoid version conflicts * enforce prores video extraction to 8 bit * make the installer more robust on execution switch * make the installer more robust on execution switch * improve the installer env handling * different approach to handle env
54 lines
1.3 KiB
Python
54 lines
1.3 KiB
Python
from typing import List
|
|
|
|
import cv2
|
|
|
|
from facefusion.types import CameraPoolSet
|
|
|
|
CAMERA_POOL_SET : CameraPoolSet =\
|
|
{
|
|
'capture': {}
|
|
}
|
|
|
|
|
|
def get_local_camera_capture(camera_id : int) -> cv2.VideoCapture:
|
|
camera_key = str(camera_id)
|
|
|
|
if camera_key not in CAMERA_POOL_SET.get('capture'):
|
|
camera_capture = cv2.VideoCapture(camera_id)
|
|
|
|
if camera_capture.isOpened():
|
|
CAMERA_POOL_SET['capture'][camera_key] = camera_capture
|
|
|
|
return CAMERA_POOL_SET.get('capture').get(camera_key)
|
|
|
|
|
|
def get_remote_camera_capture(camera_url : str) -> cv2.VideoCapture:
|
|
if camera_url not in CAMERA_POOL_SET.get('capture'):
|
|
camera_capture = cv2.VideoCapture(camera_url)
|
|
|
|
if camera_capture.isOpened():
|
|
CAMERA_POOL_SET['capture'][camera_url] = camera_capture
|
|
|
|
return CAMERA_POOL_SET.get('capture').get(camera_url)
|
|
|
|
|
|
def clear_camera_pool() -> None:
|
|
for camera_capture in CAMERA_POOL_SET.get('capture').values():
|
|
camera_capture.release()
|
|
|
|
CAMERA_POOL_SET['capture'].clear()
|
|
|
|
|
|
def detect_local_camera_ids(id_start : int, id_end : int) -> List[int]:
|
|
local_camera_ids = []
|
|
|
|
for camera_id in range(id_start, id_end):
|
|
cv2.utils.logging.setLogLevel(0)
|
|
camera_capture = get_local_camera_capture(camera_id)
|
|
cv2.utils.logging.setLogLevel(3)
|
|
|
|
if camera_capture and camera_capture.isOpened():
|
|
local_camera_ids.append(camera_id)
|
|
|
|
return local_camera_ids
|