diff --git a/README.md b/README.md index 7950c02..fb00b50 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -# Roop-GE 0.2.0 for StableDiffusion +# Roop-GE 0.2.1 for StableDiffusion ### NSFW (uncensored) version (use it on your own responsibility) of [original sd-webui-roop](https://github.com/s0md3v/sd-webui-roop) with a lot of improvements > GE (Gourieff Edition), aka "NSFW-Roop" diff --git a/scripts/roop_version.py b/scripts/roop_version.py index d369de9..6268878 100644 --- a/scripts/roop_version.py +++ b/scripts/roop_version.py @@ -1,4 +1,4 @@ -version_flag = "v0.2.0" +version_flag = "v0.2.1" from scripts.roop_logging import logger diff --git a/scripts/swapper.py b/scripts/swapper.py index a3d5e3e..3208f89 100644 --- a/scripts/swapper.py +++ b/scripts/swapper.py @@ -50,6 +50,17 @@ def cosine_similarity(test_vec: np.ndarray, source_vecs: List[np.ndarray]) -> fl FS_MODEL = None CURRENT_FS_MODEL_PATH = None +ANALYSIS_MODEL = None + + +def getAnalysisModel(): + global ANALYSIS_MODEL + if ANALYSIS_MODEL is None: + ANALYSIS_MODEL = insightface.app.FaceAnalysis( + name="buffalo_l", providers=providers + ) + return ANALYSIS_MODEL + def getFaceSwapModel(model_path: str): global FS_MODEL @@ -116,7 +127,7 @@ def upscale_image(image: Image, upscale_options: UpscaleOptions): def get_face_single(img_data: np.ndarray, face_index=0, det_size=(640, 640)): - face_analyser = insightface.app.FaceAnalysis(name="buffalo_l", providers=providers) + face_analyser = copy.deepcopy(getAnalysisModel()) face_analyser.prepare(ctx_id=0, det_size=det_size) face = face_analyser.get(img_data)