mirror of
https://github.com/facefusion/facefusion.git
synced 2026-09-16 12:35:27 +02:00
restructure config for processors
This commit is contained in:
@@ -75,17 +75,29 @@ workflow_strategy =
|
||||
|
||||
[processors]
|
||||
processors =
|
||||
|
||||
[processors.age_modifier]
|
||||
age_modifier_model =
|
||||
age_modifier_direction =
|
||||
|
||||
[processors.background_remover]
|
||||
background_remover_model =
|
||||
background_remover_fill_color =
|
||||
background_remover_despill_color =
|
||||
|
||||
[processors.deep_swapper]
|
||||
deep_swapper_model =
|
||||
deep_swapper_morph =
|
||||
|
||||
[processors.expression_restorer]
|
||||
expression_restorer_model =
|
||||
expression_restorer_factor =
|
||||
expression_restorer_areas =
|
||||
|
||||
[processors.face_debugger]
|
||||
face_debugger_items =
|
||||
|
||||
[processors.face_editor]
|
||||
face_editor_model =
|
||||
face_editor_eyebrow_direction =
|
||||
face_editor_eye_gaze_horizontal =
|
||||
@@ -101,17 +113,27 @@ face_editor_mouth_position_vertical =
|
||||
face_editor_head_pitch =
|
||||
face_editor_head_yaw =
|
||||
face_editor_head_roll =
|
||||
|
||||
[processors.face_enhancer]
|
||||
face_enhancer_model =
|
||||
face_enhancer_blend =
|
||||
face_enhancer_weight =
|
||||
|
||||
[processors.face_swapper]
|
||||
face_swapper_model =
|
||||
face_swapper_pixel_boost =
|
||||
face_swapper_weight =
|
||||
|
||||
[processors.frame_colorizer]
|
||||
frame_colorizer_model =
|
||||
frame_colorizer_size =
|
||||
frame_colorizer_blend =
|
||||
|
||||
[processors.frame_enhancer]
|
||||
frame_enhancer_model =
|
||||
frame_enhancer_blend =
|
||||
|
||||
[processors.lip_syncer]
|
||||
lip_syncer_model =
|
||||
lip_syncer_weight =
|
||||
|
||||
|
||||
@@ -130,14 +130,14 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--age-modifier-model',
|
||||
help = translator.get('help.model', __package__),
|
||||
default = config.get_str_value('processors', 'age_modifier_model', 'fran'),
|
||||
default = config.get_str_value('processors.age_modifier', 'age_modifier_model', 'fran'),
|
||||
choices = age_modifier_choices.age_modifier_models
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--age-modifier-direction',
|
||||
help = translator.get('help.direction', __package__),
|
||||
type = int,
|
||||
default = config.get_int_value('processors', 'age_modifier_direction', '0'),
|
||||
default = config.get_int_value('processors.age_modifier', 'age_modifier_direction', '0'),
|
||||
choices = age_modifier_choices.age_modifier_direction_range,
|
||||
metavar = create_int_metavar(age_modifier_choices.age_modifier_direction_range)
|
||||
)
|
||||
|
||||
@@ -501,21 +501,21 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--background-remover-model',
|
||||
help = translator.get('help.model', __package__),
|
||||
default = config.get_str_value('processors', 'background_remover_model', 'modnet'),
|
||||
default = config.get_str_value('processors.background_remover', 'background_remover_model', 'modnet'),
|
||||
choices = background_remover_choices.background_remover_models
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--background-remover-fill-color',
|
||||
help = translator.get('help.fill_color', __package__),
|
||||
type = partial(sanitize_int_range, int_range = background_remover_choices.background_remover_color_range),
|
||||
default = config.get_int_list('processors', 'background_remover_fill_color', '0 0 0 0'),
|
||||
default = config.get_int_list('processors.background_remover', 'background_remover_fill_color', '0 0 0 0'),
|
||||
nargs = '+'
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--background-remover-despill-color',
|
||||
help = translator.get('help.despill_color', __package__),
|
||||
type = partial(sanitize_int_range, int_range = background_remover_choices.background_remover_color_range),
|
||||
default = config.get_int_list('processors', 'background_remover_despill_color', '0 0 0 0'),
|
||||
default = config.get_int_list('processors.background_remover', 'background_remover_despill_color', '0 0 0 0'),
|
||||
nargs = '+'
|
||||
)
|
||||
],
|
||||
|
||||
@@ -282,14 +282,14 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--deep-swapper-model',
|
||||
help = translator.get('help.model', __package__),
|
||||
default = config.get_str_value('processors', 'deep_swapper_model', 'iperov/elon_musk_224'),
|
||||
default = config.get_str_value('processors.deep_swapper', 'deep_swapper_model', 'iperov/elon_musk_224'),
|
||||
choices = deep_swapper_choices.deep_swapper_models
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--deep-swapper-morph',
|
||||
help = translator.get('help.morph', __package__),
|
||||
type = int,
|
||||
default = config.get_int_value('processors', 'deep_swapper_morph', '100'),
|
||||
default = config.get_int_value('processors.deep_swapper', 'deep_swapper_morph', '100'),
|
||||
choices = deep_swapper_choices.deep_swapper_morph_range,
|
||||
metavar = create_int_metavar(deep_swapper_choices.deep_swapper_morph_range)
|
||||
)
|
||||
|
||||
@@ -105,21 +105,21 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--expression-restorer-model',
|
||||
help = translator.get('help.model', __package__),
|
||||
default = config.get_str_value('processors', 'expression_restorer_model', 'live_portrait'),
|
||||
default = config.get_str_value('processors.expression_restorer', 'expression_restorer_model', 'live_portrait'),
|
||||
choices = expression_restorer_choices.expression_restorer_models
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--expression-restorer-factor',
|
||||
help = translator.get('help.factor', __package__),
|
||||
type = int,
|
||||
default = config.get_int_value('processors', 'expression_restorer_factor', '80'),
|
||||
default = config.get_int_value('processors.expression_restorer', 'expression_restorer_factor', '80'),
|
||||
choices = expression_restorer_choices.expression_restorer_factor_range,
|
||||
metavar = create_int_metavar(expression_restorer_choices.expression_restorer_factor_range)
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--expression-restorer-areas',
|
||||
help = translator.get('help.areas', __package__).format(choices = ', '.join(expression_restorer_choices.expression_restorer_areas)),
|
||||
default = config.get_str_list('processors', 'expression_restorer_areas', ' '.join(expression_restorer_choices.expression_restorer_areas)),
|
||||
default = config.get_str_list('processors.expression_restorer', 'expression_restorer_areas', ' '.join(expression_restorer_choices.expression_restorer_areas)),
|
||||
choices = expression_restorer_choices.expression_restorer_areas,
|
||||
nargs = '+',
|
||||
metavar = 'EXPRESSION_RESTORER_AREAS'
|
||||
|
||||
@@ -38,7 +38,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--face-debugger-items',
|
||||
help = translator.get('help.items', __package__).format(choices = ', '.join(face_debugger_choices.face_debugger_items)),
|
||||
default = config.get_str_list('processors', 'face_debugger_items', 'face-landmark-5/68 face-mask'),
|
||||
default = config.get_str_list('processors.face_debugger', 'face_debugger_items', 'face-landmark-5/68 face-mask'),
|
||||
choices = face_debugger_choices.face_debugger_items,
|
||||
nargs = '+',
|
||||
metavar = 'FACE_DEBUGGER_ITEMS'
|
||||
|
||||
@@ -135,14 +135,14 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--face-editor-model',
|
||||
help = translator.get('help.model', __package__),
|
||||
default = config.get_str_value('processors', 'face_editor_model', 'live_portrait'),
|
||||
default = config.get_str_value('processors.face_editor', 'face_editor_model', 'live_portrait'),
|
||||
choices = face_editor_choices.face_editor_models
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--face-editor-eyebrow-direction',
|
||||
help = translator.get('help.eyebrow_direction', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_eyebrow_direction', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_eyebrow_direction', '0'),
|
||||
choices = face_editor_choices.face_editor_eyebrow_direction_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_eyebrow_direction_range)
|
||||
),
|
||||
@@ -150,7 +150,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-eye-gaze-horizontal',
|
||||
help = translator.get('help.eye_gaze_horizontal', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_eye_gaze_horizontal', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_eye_gaze_horizontal', '0'),
|
||||
choices = face_editor_choices.face_editor_eye_gaze_horizontal_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_eye_gaze_horizontal_range)
|
||||
),
|
||||
@@ -158,7 +158,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-eye-gaze-vertical',
|
||||
help = translator.get('help.eye_gaze_vertical', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_eye_gaze_vertical', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_eye_gaze_vertical', '0'),
|
||||
choices = face_editor_choices.face_editor_eye_gaze_vertical_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_eye_gaze_vertical_range)
|
||||
),
|
||||
@@ -166,7 +166,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-eye-open-ratio',
|
||||
help = translator.get('help.eye_open_ratio', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_eye_open_ratio', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_eye_open_ratio', '0'),
|
||||
choices = face_editor_choices.face_editor_eye_open_ratio_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_eye_open_ratio_range)
|
||||
),
|
||||
@@ -174,7 +174,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-lip-open-ratio',
|
||||
help = translator.get('help.lip_open_ratio', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_lip_open_ratio', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_lip_open_ratio', '0'),
|
||||
choices = face_editor_choices.face_editor_lip_open_ratio_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_lip_open_ratio_range)
|
||||
),
|
||||
@@ -182,7 +182,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-mouth-grim',
|
||||
help = translator.get('help.mouth_grim', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_mouth_grim', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_grim', '0'),
|
||||
choices = face_editor_choices.face_editor_mouth_grim_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_grim_range)
|
||||
),
|
||||
@@ -190,7 +190,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-mouth-pout',
|
||||
help = translator.get('help.mouth_pout', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_mouth_pout', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_pout', '0'),
|
||||
choices = face_editor_choices.face_editor_mouth_pout_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_pout_range)
|
||||
),
|
||||
@@ -198,7 +198,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-mouth-purse',
|
||||
help = translator.get('help.mouth_purse', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_mouth_purse', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_purse', '0'),
|
||||
choices = face_editor_choices.face_editor_mouth_purse_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_purse_range)
|
||||
),
|
||||
@@ -206,7 +206,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-mouth-smile',
|
||||
help = translator.get('help.mouth_smile', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_mouth_smile', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_smile', '0'),
|
||||
choices = face_editor_choices.face_editor_mouth_smile_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_smile_range)
|
||||
),
|
||||
@@ -214,7 +214,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-mouth-position-horizontal',
|
||||
help = translator.get('help.mouth_position_horizontal', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_mouth_position_horizontal', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_position_horizontal', '0'),
|
||||
choices = face_editor_choices.face_editor_mouth_position_horizontal_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_position_horizontal_range)
|
||||
),
|
||||
@@ -222,7 +222,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-mouth-position-vertical',
|
||||
help = translator.get('help.mouth_position_vertical', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_mouth_position_vertical', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_mouth_position_vertical', '0'),
|
||||
choices = face_editor_choices.face_editor_mouth_position_vertical_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_mouth_position_vertical_range)
|
||||
),
|
||||
@@ -230,7 +230,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-head-pitch',
|
||||
help = translator.get('help.head_pitch', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_head_pitch', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_head_pitch', '0'),
|
||||
choices = face_editor_choices.face_editor_head_pitch_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_head_pitch_range)
|
||||
),
|
||||
@@ -238,7 +238,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-head-yaw',
|
||||
help = translator.get('help.head_yaw', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_head_yaw', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_head_yaw', '0'),
|
||||
choices = face_editor_choices.face_editor_head_yaw_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_head_yaw_range)
|
||||
),
|
||||
@@ -246,7 +246,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-editor-head-roll',
|
||||
help = translator.get('help.head_roll', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_editor_head_roll', '0'),
|
||||
default = config.get_float_value('processors.face_editor', 'face_editor_head_roll', '0'),
|
||||
choices = face_editor_choices.face_editor_head_roll_range,
|
||||
metavar = create_float_metavar(face_editor_choices.face_editor_head_roll_range)
|
||||
)
|
||||
|
||||
@@ -299,14 +299,14 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--face-enhancer-model',
|
||||
help = translator.get('help.model', __package__),
|
||||
default = config.get_str_value('processors', 'face_enhancer_model', 'gfpgan_1.4'),
|
||||
default = config.get_str_value('processors.face_enhancer', 'face_enhancer_model', 'gfpgan_1.4'),
|
||||
choices = face_enhancer_choices.face_enhancer_models
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--face-enhancer-blend',
|
||||
help = translator.get('help.blend', __package__),
|
||||
type = int,
|
||||
default = config.get_int_value('processors', 'face_enhancer_blend', '80'),
|
||||
default = config.get_int_value('processors.face_enhancer', 'face_enhancer_blend', '80'),
|
||||
choices = face_enhancer_choices.face_enhancer_blend_range,
|
||||
metavar = create_int_metavar(face_enhancer_choices.face_enhancer_blend_range)
|
||||
),
|
||||
@@ -314,7 +314,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
'--face-enhancer-weight',
|
||||
help = translator.get('help.weight', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_enhancer_weight', '0.5'),
|
||||
default = config.get_float_value('processors.face_enhancer', 'face_enhancer_weight', '0.5'),
|
||||
choices = face_enhancer_choices.face_enhancer_weight_range,
|
||||
metavar = create_float_metavar(face_enhancer_choices.face_enhancer_weight_range)
|
||||
)
|
||||
|
||||
@@ -532,7 +532,7 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--face-swapper-model',
|
||||
help = translator.get('help.model', __package__),
|
||||
default = config.get_str_value('processors', 'face_swapper_model', 'hyperswap_1a_256'),
|
||||
default = config.get_str_value('processors.face_swapper', 'face_swapper_model', 'hyperswap_1a_256'),
|
||||
choices = face_swapper_choices.face_swapper_models
|
||||
)
|
||||
],
|
||||
@@ -546,14 +546,14 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--face-swapper-pixel-boost',
|
||||
help = translator.get('help.pixel_boost', __package__),
|
||||
default = config.get_str_value('processors', 'face_swapper_pixel_boost', get_first(face_swapper_pixel_boost_choices)),
|
||||
default = config.get_str_value('processors.face_swapper', 'face_swapper_pixel_boost', get_first(face_swapper_pixel_boost_choices)),
|
||||
choices = face_swapper_pixel_boost_choices
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--face-swapper-weight',
|
||||
help = translator.get('help.weight', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'face_swapper_weight', '0.5'),
|
||||
default = config.get_float_value('processors.face_swapper', 'face_swapper_weight', '0.5'),
|
||||
choices = face_swapper_choices.face_swapper_weight_range
|
||||
)
|
||||
],
|
||||
|
||||
@@ -192,21 +192,21 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--frame-colorizer-model',
|
||||
help = translator.get('help.model', __package__),
|
||||
default = config.get_str_value('processors', 'frame_colorizer_model', 'ddcolor'),
|
||||
default = config.get_str_value('processors.frame_colorizer', 'frame_colorizer_model', 'ddcolor'),
|
||||
choices = frame_colorizer_choices.frame_colorizer_models
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--frame-colorizer-size',
|
||||
help = translator.get('help.size', __package__),
|
||||
type = str,
|
||||
default = config.get_str_value('processors', 'frame_colorizer_size', '256x256'),
|
||||
default = config.get_str_value('processors.frame_colorizer', 'frame_colorizer_size', '256x256'),
|
||||
choices = frame_colorizer_choices.frame_colorizer_sizes
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--frame-colorizer-blend',
|
||||
help = translator.get('help.blend', __package__),
|
||||
type = int,
|
||||
default = config.get_int_value('processors', 'frame_colorizer_blend', '100'),
|
||||
default = config.get_int_value('processors.frame_colorizer', 'frame_colorizer_blend', '100'),
|
||||
choices = frame_colorizer_choices.frame_colorizer_blend_range,
|
||||
metavar = create_int_metavar(frame_colorizer_choices.frame_colorizer_blend_range)
|
||||
)
|
||||
|
||||
@@ -586,14 +586,14 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--frame-enhancer-model',
|
||||
help = translator.get('help.model', __package__),
|
||||
default = config.get_str_value('processors', 'frame_enhancer_model', 'span_kendata_x4'),
|
||||
default = config.get_str_value('processors.frame_enhancer', 'frame_enhancer_model', 'span_kendata_x4'),
|
||||
choices = frame_enhancer_choices.frame_enhancer_models
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--frame-enhancer-blend',
|
||||
help = translator.get('help.blend', __package__),
|
||||
type = int,
|
||||
default = config.get_int_value('processors', 'frame_enhancer_blend', '80'),
|
||||
default = config.get_int_value('processors.frame_enhancer', 'frame_enhancer_blend', '80'),
|
||||
choices = frame_enhancer_choices.frame_enhancer_blend_range,
|
||||
metavar = create_int_metavar(frame_enhancer_choices.frame_enhancer_blend_range)
|
||||
)
|
||||
|
||||
@@ -139,14 +139,14 @@ def register_args(program : ArgumentParser) -> None:
|
||||
group_processors.add_argument(
|
||||
'--lip-syncer-model',
|
||||
help = translator.get('help.model', __package__),
|
||||
default = config.get_str_value('processors', 'lip_syncer_model', 'wav2lip_gan_96'),
|
||||
default = config.get_str_value('processors.lip_syncer', 'lip_syncer_model', 'wav2lip_gan_96'),
|
||||
choices = lip_syncer_choices.lip_syncer_models
|
||||
),
|
||||
group_processors.add_argument(
|
||||
'--lip-syncer-weight',
|
||||
help = translator.get('help.weight', __package__),
|
||||
type = float,
|
||||
default = config.get_float_value('processors', 'lip_syncer_weight', '0.5'),
|
||||
default = config.get_float_value('processors.lip_syncer', 'lip_syncer_weight', '0.5'),
|
||||
choices = lip_syncer_choices.lip_syncer_weight_range,
|
||||
metavar = create_float_metavar(lip_syncer_choices.lip_syncer_weight_range)
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user