fix path for model
parent
4874282642
commit
a840986159
|
@ -11,7 +11,6 @@ from modules.face_analyser import get_one_face
|
|||
from modules.typing import Frame, Face
|
||||
from modules.utilities import (
|
||||
conditional_download,
|
||||
resolve_relative_path,
|
||||
is_image,
|
||||
is_video,
|
||||
)
|
||||
|
@ -21,9 +20,11 @@ THREAD_SEMAPHORE = threading.Semaphore()
|
|||
THREAD_LOCK = threading.Lock()
|
||||
NAME = "DLC.FACE-ENHANCER"
|
||||
|
||||
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')
|
||||
|
||||
def pre_check() -> bool:
|
||||
download_directory_path = resolve_relative_path("..\models")
|
||||
download_directory_path = models_dir
|
||||
conditional_download(
|
||||
download_directory_path,
|
||||
[
|
||||
|
@ -47,11 +48,7 @@ def get_face_enhancer() -> Any:
|
|||
|
||||
with THREAD_LOCK:
|
||||
if FACE_ENHANCER is None:
|
||||
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")
|
||||
model_path = os.path.join(models_dir, 'GFPGANv1.4.pth')
|
||||
FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1) # type: ignore[attr-defined]
|
||||
return FACE_ENHANCER
|
||||
|
||||
|
|
Loading…
Reference in New Issue