From 75c8f583550bb66d01217f803635f8063781d6e1 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 20 Jul 2023 12:46:48 +0200 Subject: [PATCH] Hotfix enhancer as of empty bbox --- roop/metadata.py | 2 +- roop/processors/frame/face_enhancer.py | 14 ++++++++------ 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/roop/metadata.py b/roop/metadata.py index 0f4e051..5870958 100644 --- a/roop/metadata.py +++ b/roop/metadata.py @@ -1,2 +1,2 @@ name = 'roop' -version = '1.2.0' +version = '1.2.1' diff --git a/roop/processors/frame/face_enhancer.py b/roop/processors/frame/face_enhancer.py index 7f9b0bb..20cdd95 100644 --- a/roop/processors/frame/face_enhancer.py +++ b/roop/processors/frame/face_enhancer.py @@ -60,12 +60,14 @@ def post_process() -> None: def enhance_face(target_face: Face, temp_frame: Frame) -> Frame: start_x, start_y, end_x, end_y = map(int, target_face['bbox']) - with THREAD_SEMAPHORE: - _, _, temp_face = get_face_enhancer().enhance( - temp_frame[start_y:end_y, start_x:end_x], - paste_back=True - ) - temp_frame[start_y:end_y, start_x:end_x] = temp_face + temp_face = temp_frame[start_y:end_y, start_x:end_x] + if temp_face.size: + with THREAD_SEMAPHORE: + _, _, temp_face = get_face_enhancer().enhance( + temp_face, + paste_back=True + ) + temp_frame[start_y:end_y, start_x:end_x] = temp_face return temp_frame