From e30f28d20e0d8c8a308dbf9cfce14bb364eea8d2 Mon Sep 17 00:00:00 2001 From: toto Date: Thu, 21 Dec 2023 12:35:35 +0800 Subject: [PATCH] update ali --- portrait/nodes.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/portrait/nodes.py b/portrait/nodes.py index fb0c1b0..fe46ed2 100644 --- a/portrait/nodes.py +++ b/portrait/nodes.py @@ -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),)