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

Compare commits

base: diaojiaolou:390b88216bd5f256276d52f08c21c6e0c7d48e37
Branches Tags
diaojiaolou:main
diaojiaolou:refactoring
diaojiaolou:testing
diaojiaolou:privacy
diaojiaolou:pr/894
diaojiaolou:UI
diaojiaolou:experimental
diaojiaolou:2.2
diaojiaolou:2.1
diaojiaolou:2.0
diaojiaolou:1.8
diaojiaolou:1.7
diaojiaolou:Deep-Live-Cam
diaojiaolou:Gumroad
diaojiaolou:Deep_Live_Cam_CUDA
..
compare: diaojiaolou:80477676b4d4edec572edd17ffdff3e15b2260b6
Branches Tags
diaojiaolou:main
diaojiaolou:refactoring
diaojiaolou:testing
diaojiaolou:privacy
diaojiaolou:pr/894
diaojiaolou:UI
diaojiaolou:experimental
diaojiaolou:2.2
diaojiaolou:2.1
diaojiaolou:2.0
diaojiaolou:1.8
diaojiaolou:1.7
diaojiaolou:Deep-Live-Cam
diaojiaolou:Gumroad
diaojiaolou:Deep_Live_Cam_CUDA

3 Commits
390b88216b ... 80477676b4

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
1 changed files with 9 additions and 2 deletions
Show Stats Expand all files Collapse all files

11
modules/processors/frame/face_swapper.py
View File

@ -4,6 +4,7 @@ import insightface
import threading import threading
import numpy as np import numpy as np
import modules.globals import modules.globals
import logging
import modules.processors.frame.core import modules.processors.frame.core
from modules.core import update_status from modules.core import update_status
from modules.face_analyser import get_one_face, get_many_faces, default_source_face 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) many_faces = get_many_faces(temp_frame)
if many_faces: if many_faces:
for target_face in many_faces: for target_face in many_faces:
temp_frame = swap_face(source_face, target_face, temp_frame) 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: else:
target_face = get_one_face(temp_frame) 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) temp_frame = swap_face(source_face, target_face, temp_frame)
else:
logging.error("Face detection failed for target or source.")
return temp_frame return temp_frame
def process_frame_v2(temp_frame: Frame, temp_frame_path: str = "") -> Frame: def process_frame_v2(temp_frame: Frame, temp_frame_path: str = "") -> Frame:
if is_image(modules.globals.target_path): if is_image(modules.globals.target_path):
if modules.globals.many_faces: if modules.globals.many_faces: