From 6dfa72314704d19c3d5b9029c34d9975ba834e74 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 17 Oct 2024 18:32:05 +0200 Subject: [PATCH] Revert back [0][0] --- facefusion/processors/modules/age_modifier.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/facefusion/processors/modules/age_modifier.py b/facefusion/processors/modules/age_modifier.py index f48c01bc..e0a78e5a 100755 --- a/facefusion/processors/modules/age_modifier.py +++ b/facefusion/processors/modules/age_modifier.py @@ -168,7 +168,7 @@ def forward(crop_vision_frame : VisionFrame, extend_vision_frame : VisionFrame) age_modifier_inputs[age_modifier_input.name] = prepare_direction(state_manager.get_item('age_modifier_direction')) with thread_semaphore(): - crop_vision_frame = age_modifier.run(None, age_modifier_inputs)[0] + crop_vision_frame = age_modifier.run(None, age_modifier_inputs)[0][0] return crop_vision_frame @@ -216,7 +216,7 @@ def normalize_extend_frame(extend_vision_frame : VisionFrame) -> VisionFrame: model_sizes = get_model_options().get('sizes') extend_vision_frame = numpy.clip(extend_vision_frame, -1, 1) extend_vision_frame = (extend_vision_frame + 1) / 2 - extend_vision_frame = extend_vision_frame[0].transpose(1, 2, 0).clip(0, 255) + extend_vision_frame = extend_vision_frame.transpose(1, 2, 0).clip(0, 255) extend_vision_frame = (extend_vision_frame * 255.0) extend_vision_frame = extend_vision_frame.astype(numpy.uint8)[:, :, ::-1] extend_vision_frame = cv2.resize(extend_vision_frame, (model_sizes.get('target')[0] * 4, model_sizes.get('target')[1] * 4), interpolation = cv2.INTER_AREA)