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 ui.py

pull/683/head
negativenagesh 2024-10-02 10:22:02 +05:30
parent 0e9ff0c54e
commit 33977b8fb7
1 changed files with 34 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
modules/ui.py
View File

@ -336,54 +336,50 @@ 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(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: Optional[pyvirtualcam.Camera] = None
) -> bool:
try: try:
return _process_webcam_frames(camera, source_image, frame_processors, virtual_cam) return _process_webcam_frames(camera, source_image, frame_processors, virtual_cam)
except Exception as e: except Exception as e:
update_status(f"Error in webcam preview: {str(e)}") logging.error(f"Error in webcam preview: {str(e)}")
return False return False
def _process_webcam_frames(camera, source_image, frame_processors, virtual_cam): def _process_webcam_frames(
global preview_label, PREVIEW camera: cv2.VideoCapture,
source_image: Any,
frame_processors: List[ModuleType],
virtual_cam: Optional[pyvirtualcam.Camera] = None
) -> bool:
while True:
ret, frame = camera.read() ret, frame = camera.read()
if not ret: if not ret:
update_status("Error: Frame not received from camera.") logging.error("Failed to read frame from camera")
return False return False
# Perform operations only if necessary # Apply any frame processors
if modules.globals.live_mirror: for processor in frame_processors:
frame = cv2.flip(frame, 1) # horizontal flipping frame = processor.process(frame, source_image)
if modules.globals.live_resizable: # Show frame preview
frame = fit_image_to_size(frame, PREVIEW.winfo_width(), PREVIEW.winfo_height()) cv2.imshow('Webcam Preview', frame)
# Process the frame using the frame processors
for frame_processor in frame_processors:
try:
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
rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
image = Image.fromarray(rgb_frame)
image = ImageOps.contain(image, (PREVIEW_MAX_WIDTH, PREVIEW_MAX_HEIGHT), Image.LANCZOS)
tk_image = ctk.CTkImage(image, size=image.size)
preview_label.configure(image=tk_image)
# Send frame to virtual camera if available
if virtual_cam: if virtual_cam:
virtual_cam.send(frame) virtual_cam.send(frame)
virtual_cam.sleep_until_next_frame() virtual_cam.sleep_until_next_frame()
ROOT.update() # Break loop on key press
if cv2.waitKey(1) & 0xFF == ord('q'):
if PREVIEW.state() == 'withdrawn': break
return False
camera.release()
cv2.destroyAllWindows()
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