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

edited utils.py

pull/676/head
negativenagesh 2024-10-02 03:00:41 +05:30
parent 615ff98e47
commit 8bc9f59ca0
1 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
modules/ui.py
View File

@ -336,6 +336,13 @@ def update_preview(frame_number: int = 0) -> None:
preview_label.configure(image=image) preview_label.configure(image=image)
def webcam_preview_loop(camera: cv2.VideoCapture, source_image: Any, frame_processors: List[ModuleType], virtual_cam: pyvirtualcam.Camera = None) -> bool: def webcam_preview_loop(camera: cv2.VideoCapture, source_image: Any, frame_processors: List[ModuleType], virtual_cam: pyvirtualcam.Camera = None) -> bool:
try:
return _process_webcam_frames(camera, source_image, frame_processors, virtual_cam)
except Exception as e:
update_status(f"Error in webcam preview: {str(e)}")
return False
def _process_webcam_frames(camera, source_image, frame_processors, virtual_cam):
global preview_label, PREVIEW global preview_label, PREVIEW
ret, frame = camera.read() ret, frame = camera.read()
@ -352,7 +359,10 @@ def webcam_preview_loop(camera: cv2.VideoCapture, source_image: Any, frame_proce
# Process the frame using the frame processors # Process the frame using the frame processors
for frame_processor in frame_processors: for frame_processor in frame_processors:
try:
frame = frame_processor.process_frame(source_image, frame) frame = frame_processor.process_frame(source_image, frame)
except Exception as e:
update_status(f"Error processing frame: {str(e)}")
# Convert to RGB only once and resize efficiently # Convert to RGB only once and resize efficiently
rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
@ -372,6 +382,7 @@ def webcam_preview_loop(camera: cv2.VideoCapture, source_image: Any, frame_proce
return True return True
def fit_image_to_size(image, width: int, height: int): def fit_image_to_size(image, width: int, height: int):
if width is None and height is None: if width is None and height is None:
return image return image
@ -428,12 +439,17 @@ def webcam_preview(camera_name: str, virtual_cam_output: bool):
preview_running = True preview_running = True
if virtual_cam_output: if virtual_cam_output:
if not PERFORMANCE_MODE:
with pyvirtualcam.Camera(width=WIDTH, height=HEIGHT, fps=FPS, fmt=pyvirtualcam.PixelFormat.BGR) as virtual_cam: with pyvirtualcam.Camera(width=WIDTH, height=HEIGHT, fps=FPS, fmt=pyvirtualcam.PixelFormat.BGR) as virtual_cam:
while preview_running: while preview_running:
preview_running = webcam_preview_loop(camera, source_image, frame_processors, virtual_cam) preview_running = webcam_preview_loop(camera, source_image, frame_processors, virtual_cam)
else:
while preview_running: while preview_running:
preview_running = webcam_preview_loop(camera, source_image, frame_processors) preview_running = webcam_preview_loop(camera, source_image, frame_processors)
else:
while preview_running:
preview_running = webcam_preview_loop(camera, source_image, frame_processors)
if camera: camera.release() if camera: camera.release()
PREVIEW.withdraw() PREVIEW.withdraw()