diaojiaolou
/
Deep-Live-Cam
mirror of https://github.com/hacksider/Deep-Live-Cam.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added virtual cam output

pull/329/head
Aleksandr Spiridonov 2024-08-15 01:31:10 -04:00
parent 3fbc1d0433
commit cd2c3c2103
2 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
modules/ui.py
View File

@ -7,6 +7,7 @@ from types import ModuleType
import cv2 import cv2
from PIL import Image, ImageOps from PIL import Image, ImageOps
from pygrabber.dshow_graph import FilterGraph from pygrabber.dshow_graph import FilterGraph
import pyvirtualcam
# Import OS-specific modules only when necessary # Import OS-specific modules only when necessary
if platform.system() == 'Darwin': # macOS if platform.system() == 'Darwin': # macOS
@ -309,7 +310,7 @@ def update_preview(frame_number: int = 0) -> None:
image = ctk.CTkImage(image, size=image.size) image = ctk.CTkImage(image, size=image.size)
preview_label.configure(image=image) preview_label.configure(image=image)
def webcam_preview_loop(cap: cv2.VideoCapture, source_image: Any, frame_processors: List[ModuleType]): def webcam_preview_loop(cap: cv2.VideoCapture, source_image: Any, frame_processors: List[ModuleType], virtual_cam: pyvirtualcam.Camera = None) -> bool:
global preview_label, PREVIEW global preview_label, PREVIEW
ret, frame = cap.read() ret, frame = cap.read()
@ -326,7 +327,14 @@ def webcam_preview_loop(cap: cv2.VideoCapture, source_image: Any, frame_processo
image = ImageOps.contain(image, (PREVIEW_MAX_WIDTH, PREVIEW_MAX_HEIGHT), Image.LANCZOS) image = ImageOps.contain(image, (PREVIEW_MAX_WIDTH, PREVIEW_MAX_HEIGHT), Image.LANCZOS)
image = ctk.CTkImage(image, size=image.size) image = ctk.CTkImage(image, size=image.size)
preview_label.configure(image=image) preview_label.configure(image=image)
if virtual_cam:
virtual_cam.send(temp_frame)
virtual_cam.sleep_until_next_frame()
ROOT.update() ROOT.update()
if PREVIEW.state() == 'withdrawn':
return False
return True return True
def webcam_preview(camera_name: str, virtual_cam_output: bool): def webcam_preview(camera_name: str, virtual_cam_output: bool):
@ -369,11 +377,15 @@ def webcam_preview(camera_name: str, virtual_cam_output: bool):
preview_running = True preview_running = True
if virtual_cam_output:
with pyvirtualcam.Camera(width=WIDTH, height=HEIGHT, fps=FPS, fmt=pyvirtualcam.PixelFormat.BGR) as virtual_cam:
while preview_running:
preview_running = webcam_preview_loop(cap, source_image, frame_processors, virtual_cam)
while preview_running: while preview_running:
preview_running = webcam_preview_loop(cap, source_image, frame_processors) preview_running = webcam_preview_loop(cap, source_image, frame_processors)
if PREVIEW.state() == 'withdrawn':
break
cap.release() cap.release()
PREVIEW.withdraw() PREVIEW.withdraw()

3
requirements.txt
View File

@ -22,4 +22,5 @@ protobuf==4.23.2
tqdm==4.66.4 tqdm==4.66.4
gfpgan==1.3.8 gfpgan==1.3.8
pyobjc==9.1; sys_platform == 'darwin' pyobjc==9.1; sys_platform == 'darwin'
pygrabber==0.2 pygrabber==0.2
pyvirtualcam==0.12.0