nodes/THtianhao-ComfyUI-Portrait-Maker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update ali

Browse Source
This commit is contained in:
toto
2023-12-21 12:35:35 +08:00
parent baa3724ae2
commit e30f28d20e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
portrait/nodes.py
+3 -3
View File
@@ -56,9 +56,9 @@ class FaceFusionPM:
def img_face_fusion(self, source_image, swap_image, mode):
if mode == "ali":
source_image = tensor_to_img(source_image)
swap_image = tensor_to_img(swap_image)
fusion_image = get_image_face_fusion()(dict(template=source_image, user=swap_image))[
source_image_pil = tensor_to_img(source_image)
swap_image_pil = tensor_to_img(swap_image)
fusion_image = get_image_face_fusion()(dict(template=source_image_pil, user=swap_image_pil))[
OutputKeys.OUTPUT_IMG]
result_image = Image.fromarray(cv2.cvtColor(fusion_image, cv2.COLOR_BGR2RGB))
return (img_to_tensor(result_image),)