apps/Deep-Live-Cam
1
0
Fork 0
mirror of https://github.com/hacksider/Deep-Live-Cam.git synced 2025-10-08 09:40:03 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Adding Pygrabber as Cam manager

Browse Source
This commit is contained in:
KRSHH
2024-12-13 19:49:11 +05:30
parent 7fb6b54c0b
commit c72582506d
6 changed files with 254 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modules/processors/frame/face_swapper.py
View File

@@ -21,7 +21,10 @@ THREAD_LOCK = threading.Lock()
NAME = "DLC.FACE-SWAPPER"
abs_dir = os.path.dirname(os.path.abspath(__file__))
models_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(abs_dir))), 'models')
models_dir = os.path.join(
os.path.dirname(os.path.dirname(os.path.dirname(abs_dir))), "models"
)
def pre_check() -> bool:
download_directory_path = abs_dir
@@ -56,7 +59,7 @@ def get_face_swapper() -> Any:
with THREAD_LOCK:
if FACE_SWAPPER is None:
model_path = os.path.join(models_dir, 'inswapper_128_fp16.onnx')
model_path = os.path.join(models_dir, "inswapper_128_fp16.onnx")
FACE_SWAPPER = insightface.model_zoo.get_model(
model_path, providers=modules.globals.execution_providers
)