1580158
/
sd-webui-roop
mirror of https://github.com/s0md3v/sd-webui-roop.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

API compatibility

pull/35/head
Somdev Sangwan 2023-06-20 21:08:36 +05:30 committed by GitHub
parent a9d53983b2
commit fc791abd29
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/swapper.py
View File

@ -111,6 +111,15 @@ def swap_face(
converted = convert_to_sd(target_img) converted = convert_to_sd(target_img)
scale, fn = converted[0], converted[1] scale, fn = converted[0], converted[1]
if model is not None and not scale: if model is not None and not scale:
if isinstance(source_img, str): # source_img is a base64 string
import base64, io
if 'base64,' in source_img: # check if the base64 string has a data URL scheme
base64_data = source_img.split('base64,')[-1]
img_bytes = base64.b64decode(base64_data)
else:
# if no data URL scheme, just decode
img_bytes = base64.b64decode(source_img)
source_img = Image.open(io.BytesIO(img_bytes))
source_img = cv2.cvtColor(np.array(source_img), cv2.COLOR_RGB2BGR) source_img = cv2.cvtColor(np.array(source_img), cv2.COLOR_RGB2BGR)
target_img = cv2.cvtColor(np.array(target_img), cv2.COLOR_RGB2BGR) target_img = cv2.cvtColor(np.array(target_img), cv2.COLOR_RGB2BGR)
source_face = get_face_single(source_img, face_index=0) source_face = get_face_single(source_img, face_index=0)