highperfocused/Deep-Live-Cam
1
0
Fork 0
mirror of https://github.com/hacksider/Deep-Live-Cam.git synced 2025-12-09 04:21:41 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

solve the FACE_ENHANCER os problem for non-nt(linux, mac) system

Browse Source
This commit is contained in:
c4fun
2024-08-07 23:26:47 +08:00
parent fc9bd973f0
commit 6400a80a91
1 changed files with 6 additions and 2 deletions
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 cv2
import threading import threading
import gfpgan import gfpgan
import os
import modules.globals import modules.globals
import modules.processors.frame.core import modules.processors.frame.core
@@ -34,8 +35,11 @@ def get_face_enhancer() -> Any:
with THREAD_LOCK: with THREAD_LOCK:
if FACE_ENHANCER is None: if FACE_ENHANCER is None:
model_path = resolve_relative_path('..\models\GFPGANv1.4.pth') if os.name == 'nt':
# todo: set models path https://github.com/TencentARC/GFPGAN/issues/399 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] FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1) # type: ignore[attr-defined]
return FACE_ENHANCER return FACE_ENHANCER