From d11a8a694104d90d899dc22da000f80670e339d7 Mon Sep 17 00:00:00 2001 From: Gourieff <777@lovemet.ru> Date: Sun, 26 Nov 2023 00:21:50 +0700 Subject: [PATCH] FIX: List index out of range when no face found Face Model Error, Issue #214 --- README.md | 2 +- README_RU.md | 2 +- scripts/reactor_swapper.py | 6 +++--- scripts/reactor_version.py | 2 +- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index 3515ce2..77635fd 100644 --- a/README.md +++ b/README.md @@ -2,7 +2,7 @@ logo - ![Version](https://img.shields.io/badge/version-0.5.1_beta3-green?style=for-the-badge&labelColor=darkgreen) + ![Version](https://img.shields.io/badge/version-0.5.1_beta4-green?style=for-the-badge&labelColor=darkgreen) Buy Me a Coffee at ko-fi.com diff --git a/README_RU.md b/README_RU.md index 93dc2f1..e0807fc 100644 --- a/README_RU.md +++ b/README_RU.md @@ -2,7 +2,7 @@ logo - ![Version](https://img.shields.io/badge/версия-0.5.1_beta3-green?style=for-the-badge&labelColor=darkgreen) + ![Version](https://img.shields.io/badge/версия-0.5.1_beta4-green?style=for-the-badge&labelColor=darkgreen) Buy Me a Coffee at ko-fi.com diff --git a/scripts/reactor_swapper.py b/scripts/reactor_swapper.py index 65d2434..69b1e62 100644 --- a/scripts/reactor_swapper.py +++ b/scripts/reactor_swapper.py @@ -533,10 +533,10 @@ def build_face_model(image: Image.Image, name: str): apply_logging_patch(1) image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR) logger.status("Building Face Model...") - face_model = analyze_faces(image)[0] - if face_model is not None: + face_model = analyze_faces(image) + if face_model is not None and len(face_model) > 0: face_model_path = os.path.join(FACE_MODELS_PATH, name + ".safetensors") - save_face_model(face_model,face_model_path) + save_face_model(face_model[0],face_model_path) logger.status("--Done!--") done_msg = f"Face model has been saved to '{face_model_path}'" logger.status(done_msg) diff --git a/scripts/reactor_version.py b/scripts/reactor_version.py index 51869bc..dda161d 100644 --- a/scripts/reactor_version.py +++ b/scripts/reactor_version.py @@ -1,5 +1,5 @@ app_title = "ReActor" -version_flag = "v0.5.1-b3" +version_flag = "v0.5.1-b4" from scripts.reactor_logger import logger, get_Run, set_Run