fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-07 18:38:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

gfpgan visibility for face_enhancer

Browse Source
This commit is contained in:
tfrymnn 2023-06-16 22:16:10 +02:00 committed by GitHub
parent f3ee39de57
commit c2db451816
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
roop/processors/frame/face_enhancer.py
View File

@ -8,6 +8,8 @@ import roop.processors.frame.core
from roop.core import update_status from roop.core import update_status
from roop.face_analyser import get_one_face, get_many_faces from roop.face_analyser import get_one_face, get_many_faces
from roop.utilities import conditional_download, resolve_relative_path, is_image, is_video from roop.utilities import conditional_download, resolve_relative_path, is_image, is_video
from PIL import Image
from numpy import asarray
FACE_ENHANCER = None FACE_ENHANCER = None
THREAD_SEMAPHORE = threading.Semaphore() THREAD_SEMAPHORE = threading.Semaphore()
@ -45,10 +47,15 @@ def get_face_enhancer() -> None:
def enhance_face(source_face: Any, target_face: Any, temp_frame: Any) -> Any: def enhance_face(source_face: Any, target_face: Any, temp_frame: Any) -> Any:
THREAD_SEMAPHORE.acquire() THREAD_SEMAPHORE.acquire()
if target_face: if target_face:
temp_frame_original = temp_frame
_, _, temp_frame = get_face_enhancer().enhance( _, _, temp_frame = get_face_enhancer().enhance(
temp_frame, temp_frame,
paste_back=True paste_back=True
) )
temp_frame_original=Image.fromarray(temp_frame_original)
temp_frame_original=temp_frame_original.resize([temp_frame_original.width*2,temp_frame_original.height*2])
temp_frame = Image.blend(temp_frame_original, Image.fromarray(temp_frame), 0.75)
temp_frame = asarray(temp_frame)
THREAD_SEMAPHORE.release() THREAD_SEMAPHORE.release()
return temp_frame return temp_frame