update roop

This commit is contained in:
toto
2023-12-21 12:39:38 +08:00
parent e30f28d20e
commit 74dce3296d
+5 -1
View File
@@ -75,6 +75,10 @@ class FaceFusionPM:
get_face_analysis().prepare(ctx_id=0, det_size=(640, 640))
faces = get_face_analysis().get(source_np)
swap_faces = get_face_analysis().get(swap_np)
if len(faces) == 0:
raise RuntimeError("No face was recognized in the source image / source image 没有识别到人脸")
if len(swap_faces) == 0:
raise RuntimeError("No face was recognized in the swap faces / swap faces没有识别到人脸")
result_image = get_roop().get(source_np, faces[0], swap_faces[0], paste_back=True)
if need_resize:
image = Image.fromarray(result_image)
@@ -494,7 +498,7 @@ class SimilarityPM:
"main_image": ("IMAGE",),
"compare_image": ("IMAGE",),
"model": (["sim"],),
"result_prefix": ("STRING",{"default": ""}),
"result_prefix": ("STRING", {"default": ""}),
},
}