Merge pull request #980 from aaddyy227/main

Fix face swapping crash due to None face embeddings
fixed import and log message
2025-03-16 00:03:39 +08:00 · 2025-03-10 23:41:28 +02:00 · 2025-03-10 23:31:56 +02:00
1 changed files with 9 additions and 2 deletions
--- a/modules/processors/frame/face_swapper.py
+++ b/modules/processors/frame/face_swapper.py
@ -4,6 +4,7 @@ import insightface
 import threading
 import numpy as np
 import modules.globals
+import logging
 import modules.processors.frame.core
 from modules.core import update_status
 from modules.face_analyser import get_one_face, get_many_faces, default_source_face
@ -105,14 +106,20 @@ def process_frame(source_face: Face, temp_frame: Frame) -> Frame:
        many_faces = get_many_faces(temp_frame)
        if many_faces:
            for target_face in many_faces:
+                if source_face and target_face:
                    temp_frame = swap_face(source_face, target_face, temp_frame)
+                else:
+                    print("Face detection failed for target/source.")
    else:
        target_face = get_one_face(temp_frame)
-        if target_face:
+        if target_face and source_face:
            temp_frame = swap_face(source_face, target_face, temp_frame)
+        else:
+            logging.error("Face detection failed for target or source.")
    return temp_frame


+
 def process_frame_v2(temp_frame: Frame, temp_frame_path: str = "") -> Frame:
    if is_image(modules.globals.target_path):
        if modules.globals.many_faces:
Author	SHA1	Message	Date
Kenneth Estanislao	80477676b4	Merge pull request #980 from aaddyy227/main Fix face swapping crash due to None face embeddings	2025-03-16 00:03:39 +08:00
Adrian Zimbran	c728994e6b	fixed import and log message	2025-03-10 23:41:28 +02:00
Adrian Zimbran	65da3be2a4	Fix face swapping crash due to None face embeddings - Add explicit checks for face detection results (source and target faces). - Handle cases when face embeddings are not available, preventing AttributeError. - Provide meaningful log messages for easier debugging in future scenarios.	2025-03-10 23:31:56 +02:00