apps/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2024-08-24 08:31:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Simplify all faces

Browse Source
This commit is contained in:
henryruhs 2023-06-03 09:37:46 +02:00
parent 6f38212bfd
commit 2a66c69d9e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
roop/swapper.py
View File

@ -29,8 +29,8 @@ def swap_face_in_frame(source_face, target_face, frame):
return frame
def process_faces(source_face, frame, progress, all_faces=False):
if all_faces:
def process_faces(source_face, frame, progress):
if roop.globals.all_faces:
many_faces = get_face_many(frame)
if many_faces:
for face in many_faces:
@ -56,7 +56,7 @@ def process_video(source_img, frame_paths):
for frame_path in frame_paths:
frame = cv2.imread(frame_path)
try:
result = process_faces(source_face, frame, progress, roop.globals.all_faces)
result = process_faces(source_face, frame, progress)
cv2.imwrite(frame_path, result)
except Exception:
progress.set_postfix(status='E', refresh=True)