Upload 2 files

roop/processors/frame/face_swapper.py

@@ -0,0 +1,23 @@
+from typing import Any
+import insightface
+import threading
+
+import roop.globals
+from roop.utilities import resolve_relative_path
+
+FACE_SWAPPER = None
+THREAD_LOCK = threading.Lock()
+NAME = 'ROOP.FACE-SWAPPER'
+
+DIST_THRESHOLD = 0.65
+
+
+def get_face_swapper() -> Any:
+    global FACE_SWAPPER
+
+    with THREAD_LOCK:
+        if FACE_SWAPPER is None:
+            model_path = resolve_relative_path('../models/inswapper_128.onnx')
+            FACE_SWAPPER = insightface.model_zoo.get_model(model_path, providers=roop.globals.execution_providers)
+    return FACE_SWAPPER
+