Recrosoftware/sd-webui-reactor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

UPDATE: det_size with face model build

Browse Source 
FR #247
This commit is contained in:
Art Gourieff 2024-02-20 19:19:05 +07:00
parent 96d7a06291
commit 9778e4f208
4 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -2,7 +2,7 @@
<img src="https://github.com/Gourieff/Assets/raw/main/sd-webui-reactor/ReActor_logo_NEW_EN.png?raw=true" alt="logo" width="180px"/>
![Version](https://img.shields.io/badge/version-0.7.0_alpha1-lightgreen?style=for-the-badge&labelColor=darkgreen)
![Version](https://img.shields.io/badge/version-0.7.0_alpha2-lightgreen?style=for-the-badge&labelColor=darkgreen)
<a href="https://boosty.to/artgourieff" target="_blank">
<img src="https://lovemet.ru/www/boosty.jpg" width="108" alt="Support Me on Boosty"/>

2
README_RU.md
View File

@ -2,7 +2,7 @@
<img src="https://github.com/Gourieff/Assets/raw/main/sd-webui-reactor/ReActor_logo_NEW_RU.png?raw=true" alt="logo" width="180px"/>
![Version](https://img.shields.io/badge/версия-0.7.0_alpha1-lightgreen?style=for-the-badge&labelColor=darkgreen)
![Version](https://img.shields.io/badge/версия-0.7.0_alpha2-lightgreen?style=for-the-badge&labelColor=darkgreen)
<a href="https://boosty.to/artgourieff" target="_blank">
<img src="https://lovemet.ru/www/boosty.jpg" width="108" alt="Поддержать проект на Boosty"/>

9
scripts/reactor_swapper.py
View File

@ -626,7 +626,7 @@ def swap_face(
return result_image, [], 0
def build_face_model(image: Image.Image, name: str, save_model: bool = True):
def build_face_model(image: Image.Image, name: str, save_model: bool = True, det_size=(640, 640)):
if image is None:
error_msg = "Please load an Image"
logger.error(error_msg)
@ -639,7 +639,12 @@ def build_face_model(image: Image.Image, name: str, save_model: bool = True):
image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
if save_model:
logger.status("Building Face Model...")
face_model = analyze_faces(image)
face_model = analyze_faces(image, det_size)
if len(face_model) == 0:
det_size_half = half_det_size(det_size)
face_model = analyze_faces(image, det_size_half)
if face_model is not None and len(face_model) > 0:
if save_model:
face_model_path = os.path.join(FACE_MODELS_PATH, name + ".safetensors")

2
scripts/reactor_version.py
View File

@ -1,5 +1,5 @@
app_title = "ReActor"
version_flag = "v0.7.0-a1"
version_flag = "v0.7.0-a2"
from scripts.reactor_logger import logger, get_Run, set_Run
from scripts.reactor_globals import DEVICE