From 4322a992e9f05d7488128b83ac327a4817c0f629 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Thu, 19 Sep 2024 15:26:59 +0530 Subject: [PATCH] change expression restorer 0-100 range --- facefusion/processors/choices.py | 2 +- facefusion/processors/modules/expression_restorer.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/facefusion/processors/choices.py b/facefusion/processors/choices.py index 59367ac7..1b516ced 100755 --- a/facefusion/processors/choices.py +++ b/facefusion/processors/choices.py @@ -26,7 +26,7 @@ frame_enhancer_models : List[FrameEnhancerModel] = [ 'clear_reality_x4', 'lsdir_ lip_syncer_models : List[LipSyncerModel] = [ 'wav2lip', 'wav2lip_gan' ] age_modifier_direction_range : Sequence[int] = create_int_range(-100, 100, 1) -expression_restorer_factor_range : Sequence[int] = create_int_range(0, 120, 1) +expression_restorer_factor_range : Sequence[int] = create_int_range(0, 100, 1) face_editor_eyebrow_direction_range : Sequence[float] = create_float_range(-1.0, 1.0, 0.05) face_editor_eye_gaze_horizontal_range : Sequence[float] = create_float_range(-1.0, 1.0, 0.05) face_editor_eye_gaze_vertical_range : Sequence[float] = create_float_range(-1.0, 1.0, 0.05) diff --git a/facefusion/processors/modules/expression_restorer.py b/facefusion/processors/modules/expression_restorer.py index dbdb4f08..c432b082 100755 --- a/facefusion/processors/modules/expression_restorer.py +++ b/facefusion/processors/modules/expression_restorer.py @@ -91,7 +91,7 @@ def register_args(program : ArgumentParser) -> None: group_processors = find_argument_group(program, 'processors') if group_processors: group_processors.add_argument('--expression-restorer-model', help = wording.get('help.expression_restorer_model'), default = config.get_str_value('processors.expression_restorer_model', 'live_portrait'), choices = processors_choices.expression_restorer_models) - group_processors.add_argument('--expression-restorer-factor', help = wording.get('help.expression_restorer_factor'), type = int, default = config.get_int_value('processors.expression_restorer_factor', '100'), choices = processors_choices.expression_restorer_factor_range, metavar = create_int_metavar(processors_choices.expression_restorer_factor_range)) + group_processors.add_argument('--expression-restorer-factor', help = wording.get('help.expression_restorer_factor'), type = int, default = config.get_int_value('processors.expression_restorer_factor', '80'), choices = processors_choices.expression_restorer_factor_range, metavar = create_int_metavar(processors_choices.expression_restorer_factor_range)) facefusion.jobs.job_store.register_step_keys([ 'expression_restorer_model','expression_restorer_factor' ]) @@ -137,7 +137,7 @@ def post_process() -> None: def restore_expression(source_vision_frame : VisionFrame, target_face : Face, temp_vision_frame : VisionFrame) -> VisionFrame: model_template = get_model_options().get('template') model_size = get_model_options().get('size') - expression_restorer_factor = float(numpy.interp(float(state_manager.get_item('expression_restorer_factor')), [ 0, 120 ], [ 0, 1.2 ])) + expression_restorer_factor = float(numpy.interp(float(state_manager.get_item('expression_restorer_factor')), [ 0, 100 ], [ 0, 1.2 ])) source_vision_frame = cv2.resize(source_vision_frame, temp_vision_frame.shape[:2][::-1]) source_crop_vision_frame, _ = warp_face_by_face_landmark_5(source_vision_frame, target_face.landmark_set.get('5/68'), model_template, model_size) target_crop_vision_frame, affine_matrix = warp_face_by_face_landmark_5(temp_vision_frame, target_face.landmark_set.get('5/68'), model_template, model_size)