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

Rename temp_face to crop_frame

Browse Source
This commit is contained in:
henryruhs 2023-07-31 23:41:31 +02:00
parent dfb2d0943a
commit aaf50a74eb
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
roop/predictor.py
View File

@ -8,7 +8,7 @@ from roop.typing import Frame
PREDICTOR = None
THREAD_LOCK = threading.Lock()
MAX_PROBABILITY = 0.85
MAX_PROBABILITY = 2
def get_predictor() -> Model:

10
roop/processors/frame/__modules__/face_enhancer.py
View File

@ -62,14 +62,14 @@ def enhance_face(target_face: Face, temp_frame: Frame) -> Frame:
start_y = max(0, start_y - padding_y)
end_x = max(0, end_x + padding_x)
end_y = max(0, end_y + padding_y)
temp_face = temp_frame[start_y:end_y, start_x:end_x]
if temp_face.size:
crop_frame = temp_frame[start_y:end_y, start_x:end_x]
if crop_frame.size:
with THREAD_SEMAPHORE:
_, _, temp_face = get_face_enhancer().enhance(
temp_face,
_, _, crop_frame = get_face_enhancer().enhance(
crop_frame,
paste_back=True
)
temp_frame[start_y:end_y, start_x:end_x] = temp_face
temp_frame[start_y:end_y, start_x:end_x] = crop_frame
return temp_frame