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

Merge pull request #74 from c4fun/main

Browse Source 
solve the FACE_ENHANCER os problem for non-nt(linux, mac) system
This commit is contained in:
Kenneth Estanislao 2024-08-08 13:15:57 +08:00 committed by GitHub
commit 6bb9e1ba15
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/processors/frame/face_enhancer.py
View File

@ -2,6 +2,7 @@ from typing import Any, List
import cv2
import threading
import gfpgan
import os
import modules.globals
import modules.processors.frame.core
@ -34,8 +35,11 @@ def get_face_enhancer() -> Any:
with THREAD_LOCK:
if FACE_ENHANCER is None:
model_path = resolve_relative_path('..\models\GFPGANv1.4.pth')
# todo: set models path https://github.com/TencentARC/GFPGAN/issues/399
if os.name == 'nt':
model_path = resolve_relative_path('..\models\GFPGANv1.4.pth')
# todo: set models path https://github.com/TencentARC/GFPGAN/issues/399
else:
model_path = resolve_relative_path('../models/GFPGANv1.4.pth')
FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1) # type: ignore[attr-defined]
return FACE_ENHANCER