upagge/roop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update processor.py

Browse Source
This commit is contained in:
Henry Ruhs 2023-05-31 02:44:58 +02:00 committed by GitHub
parent 251fde9596
commit 6258f3c084
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
core/processor.py
View File

@ -37,6 +37,6 @@ def process_img(source_img, target_path, output_file):
frame = cv2.imread(target_path) frame = cv2.imread(target_path)
face = get_face(frame) face = get_face(frame)
source_face = get_face(cv2.imread(source_img)) source_face = get_face(cv2.imread(source_img))
result = face_swapper.get(frame, face, source_face, paste_back=True) result = get_face_swapper().get(frame, face, source_face, paste_back=True)
cv2.imwrite(output_file, result) cv2.imwrite(output_file, result)
print("\n\nImage saved as:", output_file, "\n\n") print("\n\nImage saved as:", output_file, "\n\n")