apps/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-10-06 00:37:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

properly name output file when its an image

Browse Source
This commit is contained in:
Somdev Sangwan
2023-05-30 19:07:22 +05:30
parent 03ac74edac
commit 90e2ba0fc7
2 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
core/processor.py
View File

@@ -28,11 +28,10 @@ def process_video(source_img, frame_paths):
pass
def process_img(source_img, target_path):
def process_img(source_img, target_path, output_file):
frame = cv2.imread(target_path)
face = get_face(frame)
source_face = get_face(cv2.imread(source_img))
result = face_swapper.get(frame, face, source_face, paste_back=True)
target_path = rreplace(target_path, "/", "/swapped-", 1) if "/" in target_path else "swapped-"+target_path
print(target_path)
cv2.imwrite(target_path, result)
cv2.imwrite(output_file, result)
print("\n\nImage saved as:", output_file, "\n\n")