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

FIX: code optimization

Browse Source
This commit is contained in:
Art Gourieff 2023-06-28 19:56:08 +07:00 committed by GitHub
parent 4cb73f597a
commit c0e8618135
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
scripts/swapper.py
View File

@ -46,18 +46,6 @@ def cosine_similarity(test_vec: np.ndarray, source_vecs: List[np.ndarray]) -> fl
return average_cos_dist 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 FS_MODEL = None
CURRENT_FS_MODEL_PATH = 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)): 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_analyser.prepare(ctx_id=0, det_size=det_size)
face = face_analyser.get(img_data) face = face_analyser.get(img_data)