From c0e8618135a67574a31c9805ec5dece5e00aef0b Mon Sep 17 00:00:00 2001 From: Art Gourieff <85128026+Gourieff@users.noreply.github.com> Date: Wed, 28 Jun 2023 19:56:08 +0700 Subject: [PATCH] FIX: code optimization --- scripts/swapper.py | 14 +------------- 1 file changed, 1 insertion(+), 13 deletions(-) diff --git a/scripts/swapper.py b/scripts/swapper.py index f1d8a0b..e76bdd5 100644 --- a/scripts/swapper.py +++ b/scripts/swapper.py @@ -46,18 +46,6 @@ def cosine_similarity(test_vec: np.ndarray, source_vecs: List[np.ndarray]) -> fl return average_cos_dist -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 - - FS_MODEL = None CURRENT_FS_MODEL_PATH = None @@ -103,7 +91,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 = copy.deepcopy(getAnalysisModel()) + face_analyser = insightface.app.FaceAnalysis(name="buffalo_l", providers=providers) face_analyser.prepare(ctx_id=0, det_size=det_size) face = face_analyser.get(img_data)